Update README.md
Browse files
README.md
CHANGED
@@ -36,8 +36,8 @@ Important points like these help models (even smaller models like 7b) to perform
|
|
36 |
!pip install -q -U bitsandbytes
|
37 |
|
38 |
from transformers import AutoModelForCausalLM, AutoTokenizer, TextStreamer
|
39 |
-
model = AutoModelForCausalLM.from_pretrained("rhaymison/luana-portuguese-7b-health", device_map= {"": 0})
|
40 |
-
tokenizer = AutoTokenizer.from_pretrained("rhaymison/luana-portuguese-7b-health")
|
41 |
model.eval()
|
42 |
|
43 |
```
|
|
|
36 |
!pip install -q -U bitsandbytes
|
37 |
|
38 |
from transformers import AutoModelForCausalLM, AutoTokenizer, TextStreamer
|
39 |
+
model = AutoModelForCausalLM.from_pretrained("rhaymison/Mistral-luana-portuguese-7b-health", device_map= {"": 0})
|
40 |
+
tokenizer = AutoTokenizer.from_pretrained("rhaymison/Mistral-luana-portuguese-7b-health")
|
41 |
model.eval()
|
42 |
|
43 |
```
|