Update README.md
Browse files
README.md
CHANGED
@@ -184,8 +184,8 @@ Important points like these help models (even smaller models like 7b) to perform
|
|
184 |
!pip install -q -U bitsandbytes
|
185 |
|
186 |
from transformers import AutoModelForCausalLM, AutoTokenizer, TextStreamer
|
187 |
-
model = AutoModelForCausalLM.from_pretrained("rhaymison/Mistral-
|
188 |
-
tokenizer = AutoTokenizer.from_pretrained("rhaymison/Mistral-
|
189 |
model.eval()
|
190 |
|
191 |
```
|
|
|
184 |
!pip install -q -U bitsandbytes
|
185 |
|
186 |
from transformers import AutoModelForCausalLM, AutoTokenizer, TextStreamer
|
187 |
+
model = AutoModelForCausalLM.from_pretrained("rhaymison/Mistral-8x7b-Quantized-portuguese-luana", device_map= {"": 0})
|
188 |
+
tokenizer = AutoTokenizer.from_pretrained("rhaymison/Mistral-8x7b-Quantized-portuguese-luana")
|
189 |
model.eval()
|
190 |
|
191 |
```
|