Commit
•
75a0869
1
Parent(s):
dbfbe5c
Update README.md
Browse files
README.md
CHANGED
@@ -99,9 +99,9 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
|
|
99 |
|
100 |
model_id = "shenzhi-wang/Llama3-8B-Chinese-Chat"
|
101 |
|
102 |
-
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
103 |
model = AutoModelForCausalLM.from_pretrained(
|
104 |
-
model_id, torch_dtype="auto", device_map="auto"
|
105 |
)
|
106 |
|
107 |
messages = [
|
|
|
99 |
|
100 |
model_id = "shenzhi-wang/Llama3-8B-Chinese-Chat"
|
101 |
|
102 |
+
tokenizer = AutoTokenizer.from_pretrained(model_id, revision="v1")
|
103 |
model = AutoModelForCausalLM.from_pretrained(
|
104 |
+
model_id, torch_dtype="auto", device_map="auto", revision="v1"
|
105 |
)
|
106 |
|
107 |
messages = [
|