Spaces:

patpizio
/

llm-token-probs

Sleeping

patpizio commited on Jul 9, 2023

Commit

74fd993

1 Parent(s): 5287368

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -49,7 +49,7 @@ if instruction:
         )
     output_text = tokenizer.decode(
-        outputs['sequences'][0].cuda(),
         skip_special_tokens=False
     ).strip()

         )
     output_text = tokenizer.decode(
+        outputs['sequences'][0],#.cuda(),
         skip_special_tokens=False
     ).strip()