htliu commited on
Commit
52ecb99
1 Parent(s): 6df5440

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -42,7 +42,7 @@ XuanYuan-6B基座模型、chat模型及其量化模型的使用方法和[XuanYua
42
  import torch
43
  from transformers import LlamaForCausalLM, AutoTokenizer
44
 
45
- model_name_or_path = "Duxiaoman-DI/XuanYuan-70B-Chat"
46
 
47
  tokenizer = AutoTokenizer.from_pretrained(model_name_or_path)
48
  model = LlamaForCausalLM.from_pretrained(model_name_or_path, device_map="auto")
 
42
  import torch
43
  from transformers import LlamaForCausalLM, AutoTokenizer
44
 
45
+ model_name_or_path = "Duxiaoman-DI/XuanYuan-6B-Chat"
46
 
47
  tokenizer = AutoTokenizer.from_pretrained(model_name_or_path)
48
  model = LlamaForCausalLM.from_pretrained(model_name_or_path, device_map="auto")