Spaces:

Wonder-Griffin
/

Testes

Runtime error

Wonder-Griffin commited on Sep 7, 2024

Commit

574cfd3

verified ·

1 Parent(s): 3d2f64c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -39,7 +39,7 @@ def respond(
     # Generate a response from the model
     output_ids = model.generate(
-        input_ids.to("cuda"),
         max_new_tokens=max_tokens,
         temperature=temperature,
         top_p=top_p,

     # Generate a response from the model
     output_ids = model.generate(
+        input_ids,
         max_new_tokens=max_tokens,
         temperature=temperature,
         top_p=top_p,