Spaces:

IkIzma
/

Transformer_Homework

Runtime error

IkIzma commited on Apr 16, 2023

Commit

63efcfa

1 Parent(s): 361d939

Update Generate_text.py

Files changed (1) hide show

Generate_text.py CHANGED Viewed

@@ -18,7 +18,7 @@ def generate(
     ):
     input_ids = tok.encode(text, return_tensors="pt").to(device)
     out = model.generate(
-        input_ids.cuda(),
         max_length=max_length,
         repetition_penalty=repetition_penalty,
         do_sample=do_sample,

     ):
     input_ids = tok.encode(text, return_tensors="pt").to(device)
     out = model.generate(
+        input_ids.to(device),
         max_length=max_length,
         repetition_penalty=repetition_penalty,
         do_sample=do_sample,