mxmax
/

Chinese_Chat_T5_Base

text2text-generation

Text2Text-Generation

text-generation-inference

Model card Files Files and versions

mxmax commited on Mar 28, 2023

Commit

5f6002d

·

1 Parent(s): 95bba7a

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -58,7 +58,7 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 tokenizer = AutoTokenizer.from_pretrained("mxmax/Chinese_Chat_T5_Base")
 model = AutoModelForSeq2SeqLM.from_pretrained("mxmax/Chinese_Chat_T5_Base")
 device = 'cuda' if cuda.is_available() else 'cpu'
-model_trained.to(device)
 def postprocess(text):
   return text.replace(".", "").replace('</>','')

 tokenizer = AutoTokenizer.from_pretrained("mxmax/Chinese_Chat_T5_Base")
 model = AutoModelForSeq2SeqLM.from_pretrained("mxmax/Chinese_Chat_T5_Base")
 device = 'cuda' if cuda.is_available() else 'cpu'
+model.to(device)
 def postprocess(text):
   return text.replace(".", "").replace('</>','')