JustinLin610 commited on
Commit
afcf99b
1 Parent(s): 2d5d47e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -42,7 +42,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
42
  device = "cuda" # the device to load the model onto
43
 
44
  model = AutoModelForCausalLM.from_pretrained(
45
- "Qwen/Qwen2-MoE-57B-A14B-Instruct",
46
  torch_dtype="auto",
47
  device_map="auto"
48
  )
 
42
  device = "cuda" # the device to load the model onto
43
 
44
  model = AutoModelForCausalLM.from_pretrained(
45
+ "Qwen/Qwen2-57B-A14B-Instruct",
46
  torch_dtype="auto",
47
  device_map="auto"
48
  )