more fix

KonferCA · Nov 24, 2024 · 615f697 · 615f697
1 parent f50b479
commit 615f697
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/backend/generate_answer.py b/backend/generate_answer.py
@@ -56,7 +56,7 @@ def generate_initial_note(page_content, model, tokenizer):
     )
     final_output = ""
     for output in outputs:
-        final_output += tokenizer.decode(output, skip_special_tokens=True).to(device)
+        final_output += tokenizer.decode(output, skip_special_tokens=True)
     return final_output
 
 def generate_note(page_content, note_content, model, tokenizer):