Merge pull request tloen#6 from janmaltel/janmaltel/input-bug

Fix bug in generate promp using 'instruction' instead of 'input'
NanQiAi · Mar 14, 2023 · 1193c63 · 1193c63
2 parents 6f46581 + a3b80fd
commit 1193c63
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/finetune.py b/finetune.py
@@ -37,7 +37,7 @@
 
 def generate_prompt(data_point):
     # sorry about the formatting disaster gotta move fast
-    if data_point["instruction"]:
+    if data_point["input"]:
         return f"""Below is an instruction that describes a task, paired with an input that provides further context. Write a response that appropriately completes the request.
 
 ### Instruction: