Spaces:

Charles1973
/

NatsumeSousekiSpace

Runtime error

Charles1973 commited on Aug 16, 2023

Commit

5bc68e8

•

1 Parent(s): e743472

修正

Files changed (1) hide show

app.py CHANGED Viewed

@@ -35,7 +35,7 @@ model.eval()
 #    return tokenizer.batch_decode(output)
 def Chat(prompt):
   num = 3
-  input_ids = tokenizer.encode(prompt, return_tensors="pt",add_special_tokens=False).to(device)
   #with torch.no_grad():
   output = model.generate(
         input_ids,

 #    return tokenizer.batch_decode(output)
 def Chat(prompt):
   num = 3
+  input_ids = tokenizer.encode(prompt, return_tensors="pt",add_special_tokens=False)
   #with torch.no_grad():
   output = model.generate(
         input_ids,