rhaymison
/

Mistral-portuguese-luana-7b-mental-health

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

rhaymison commited on Apr 4

Commit

f4ddca8

•

1 Parent(s): 939553d

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -36,8 +36,8 @@ Important points like these help models (even smaller models like 7b) to perform
 !pip install -q -U bitsandbytes
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextStreamer
-model = AutoModelForCausalLM.from_pretrained("rhaymison/luana-portuguese-7b-health", device_map= {"": 0})
-tokenizer = AutoTokenizer.from_pretrained("rhaymison/luana-portuguese-7b-health")
 model.eval()
 ```

 !pip install -q -U bitsandbytes
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextStreamer
+model = AutoModelForCausalLM.from_pretrained("rhaymison/Mistral-luana-portuguese-7b-health", device_map= {"": 0})
+tokenizer = AutoTokenizer.from_pretrained("rhaymison/Mistral-luana-portuguese-7b-health")
 model.eval()
 ```