Update fine_tuning.py

fine_tuning.py

@@ -162,7 +162,7 @@ def load_and_train(model_id="TinyLlama/TinyLlama-1.1B-Chat-v1.0"):
 # -----------------------------
 # GENERATE ANSWER
 # -----------------------------
-def generate_answer(tokenizer, device, prompt, max_tokens=200):
+def generate_answer(prompt, max_tokens=200):
     if prompt.strip() == "":
         return "Please enter a prompt!"