Update README.md
Browse files
README.md
CHANGED
@@ -182,8 +182,8 @@ Here provides a code snippet to show you how to load the tokenizer and model and
|
|
182 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
183 |
device = "cuda" # the device to load the model
|
184 |
|
185 |
-
model = AutoModelForCausalLM.from_pretrained("sail/Sailor-
|
186 |
-
tokenizer = AutoTokenizer.from_pretrained("sail/Sailor-
|
187 |
|
188 |
input_message = "Model bahasa adalah model probabilistik"
|
189 |
### The given Indonesian input translates to 'A language model is a probabilistic model of.'
|
|
|
182 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
183 |
device = "cuda" # the device to load the model
|
184 |
|
185 |
+
model = AutoModelForCausalLM.from_pretrained("sail/Sailor-0.5B", device_map="auto")
|
186 |
+
tokenizer = AutoTokenizer.from_pretrained("sail/Sailor-0.5B")
|
187 |
|
188 |
input_message = "Model bahasa adalah model probabilistik"
|
189 |
### The given Indonesian input translates to 'A language model is a probabilistic model of.'
|