Update model_class.py

model_class.py

@@ -181,7 +181,8 @@ class CustomGPT(
             f"Write a response that appropriately completes the request."
             f"\n\n### Instruction:\n{entry['instruction']}"
         )
-        input_text = f"\n\n### Input:\n{entry['input']}" if entry["input"] else ""
+        # input_text = f"\n\n### Input:\n{entry['input']}" if entry["input"] else ""
+        input_text = f"\n\n### Input:\n{entry.get('input', '')}"
         return instruction_text + input_text
 
     def text_to_token_ids(self, text, tokenizer):