ClueAI commited on
Commit
0e0b4e5
1 Parent(s): 5348be3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -80,9 +80,9 @@ import os
80
  # 加载模型
81
 
82
  model_dir='ClueAI/ChatYuan-large-v2'
83
- tokenizer = T5Tokenizer.from_pretrained(model_dir)
84
  # 速度会受到网络影响
85
- model = T5ForConditionalGeneration.from_pretrained(model_dir, trust_remote_code=True)
86
  history = []
87
  print("starting")
88
  while True:
 
80
  # 加载模型
81
 
82
  model_dir='ClueAI/ChatYuan-large-v2'
83
+ tokenizer = AutoTokenizer.from_pretrained(model_dir)
84
  # 速度会受到网络影响
85
+ model = AutoModel.from_pretrained(model_dir, trust_remote_code=True)
86
  history = []
87
  print("starting")
88
  while True: