Update README.md
Browse files
README.md
CHANGED
@@ -24,7 +24,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
|
|
24 |
import math
|
25 |
|
26 |
## v2 models
|
27 |
-
model_path = "
|
28 |
|
29 |
tokenizer = AutoTokenizer.from_pretrained(model_path, use_default_system_prompt=False)
|
30 |
model = AutoModelForCausalLM.from_pretrained(
|
@@ -50,7 +50,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
|
|
50 |
import math
|
51 |
|
52 |
## v2 models
|
53 |
-
model_path = "
|
54 |
|
55 |
tokenizer = AutoTokenizer.from_pretrained(model_path, use_default_system_prompt=False)
|
56 |
model = AutoModelForCausalLM.from_pretrained(
|
|
|
24 |
import math
|
25 |
|
26 |
## v2 models
|
27 |
+
model_path = "cloudyu/Yi-9Bx2-MOE"
|
28 |
|
29 |
tokenizer = AutoTokenizer.from_pretrained(model_path, use_default_system_prompt=False)
|
30 |
model = AutoModelForCausalLM.from_pretrained(
|
|
|
50 |
import math
|
51 |
|
52 |
## v2 models
|
53 |
+
model_path = "cloudyu/Yi-9Bx2-MOE"
|
54 |
|
55 |
tokenizer = AutoTokenizer.from_pretrained(model_path, use_default_system_prompt=False)
|
56 |
model = AutoModelForCausalLM.from_pretrained(
|