ClueAI
/

ChatYuan-large-v2

text2text-generation

text-generation-inference

Model card Files Files and versions

ClueAI commited on Jul 9, 2023

Commit

0e0b4e5

·

1 Parent(s): 5348be3

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -80,9 +80,9 @@ import os
 # 加载模型
 model_dir='ClueAI/ChatYuan-large-v2'
-tokenizer = T5Tokenizer.from_pretrained(model_dir)
 # 速度会受到网络影响
-model = T5ForConditionalGeneration.from_pretrained(model_dir, trust_remote_code=True)
 history = []
 print("starting")
 while True:

 # 加载模型
 model_dir='ClueAI/ChatYuan-large-v2'
+tokenizer = AutoTokenizer.from_pretrained(model_dir)
 # 速度会受到网络影响
+model = AutoModel.from_pretrained(model_dir, trust_remote_code=True)
 history = []
 print("starting")
 while True: