From 6463ab322434f01557cc4a55ca9da44a0b561999 Mon Sep 17 00:00:00 2001
From: "l.gabrysiak" <l.gabrysiak@osadkowski.pl>
Date: Tue, 25 Feb 2025 19:47:34 +0100
Subject: [PATCH] mod

---
 hft.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/hft.py b/hft.py
index 72ad4c0..a947664 100644
--- a/hft.py
+++ b/hft.py
@@ -192,7 +192,7 @@ trainer.train()
 # Funkcja generująca odpowiedź
 def generate_answer(question, max_length=200):
     model.eval()
-    inputs = tokenizer(question, return_tensors="pt", truncation=True, max_length=512).to(device)
+    inputs = tokenizer(question, return_tensors="pt", truncation=True, max_length=512)
     
     with torch.no_grad():
         outputs = model.generate(