소스 검색

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

Fix bug in generate promp using 'instruction' instead of 'input'
Eric J. Wang 3 년 전
부모
커밋
1193c63833
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      finetune.py

+ 1 - 1
finetune.py

@@ -37,7 +37,7 @@ data = load_dataset("json", data_files="alpaca_data.json")
 
 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: