yutaozhu94 commited on
Commit
604326f
1 Parent(s): 8436489

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -99,8 +99,8 @@ As our model is trained based on LLaMA, it can be loaded in the same way as orig
99
 
100
  ```Python
101
  >>> from transformers import LlamaTokenizer, LlamaModelForCausalLM
102
- >>> tokenizer = AutoTokenizer.from_pretrained("yulan-team/YuLan-Chat-2-13b")
103
- >>> model = AutoModelForCausalLM.from_pretrained("yulan-team/YuLan-Chat-2-13b").cuda()
104
  >>> model = model.eval()
105
  >>> input_text = "hello"
106
  >>> prompt = "The following is a conversation between a human and an AI assistant namely YuLan, developed by GSAI, Renmin University of China. The AI assistant gives helpful, detailed, and polite answers to the user's questions.\n[|Human|]:{}\n[|AI|]:".format(input_text)
 
99
 
100
  ```Python
101
  >>> from transformers import LlamaTokenizer, LlamaModelForCausalLM
102
+ >>> tokenizer = LlamaTokenizer.from_pretrained("yulan-team/YuLan-Chat-2-13b")
103
+ >>> model = LlamaModelForCausalLM.from_pretrained("yulan-team/YuLan-Chat-2-13b").cuda()
104
  >>> model = model.eval()
105
  >>> input_text = "hello"
106
  >>> prompt = "The following is a conversation between a human and an AI assistant namely YuLan, developed by GSAI, Renmin University of China. The AI assistant gives helpful, detailed, and polite answers to the user's questions.\n[|Human|]:{}\n[|AI|]:".format(input_text)