scutcyr
/

BianQue-1.0

text2text-generation

text-generation-inference

Model card Files Files and versions

scutcyr commited on Jun 6, 2023

Commit

ff60a23

·

1 Parent(s): 1f104d0

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -81,7 +81,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 tokenizer = T5Tokenizer.from_pretrained("scutcyr/BianQue-1.0")
 model = T5ForConditionalGeneration.from_pretrained("scutcyr/BianQue-1.0")
 def preprocess(text):
     text = text.replace("\n", "\\n").replace("\t", "\\t")

 tokenizer = T5Tokenizer.from_pretrained("scutcyr/BianQue-1.0")
 model = T5ForConditionalGeneration.from_pretrained("scutcyr/BianQue-1.0")
+model = model.to(device)
 def preprocess(text):
     text = text.replace("\n", "\\n").replace("\t", "\\t")