fix(datagen)

UniverseFly · UniverseFly · commit 5683e3eabc58 · 2024-05-09T11:16:38.000-05:00
diff --git a/src/star_align/self_ossinstruct.py b/src/star_align/self_ossinstruct.py
@@ -261,7 +261,7 @@ def random_prompt(
         assert len(examples) == num_fewshots
 
         body = "\n\n".join(
-            f"## Example {idx + 1}\n{example.prompt(mode, index=idx + 1)}"
+            f"## Example {idx + 1}\n{example.prompt(mode, index=idx + 1 if LLAMA3 and mode == 'I->R' else None)}"
             for idx, example in enumerate(examples)
         )
         # content = f"{self.system_prompt}\n\n{body}"

Original file line number	Diff line number	Diff line change
`@@ -261,7 +261,7 @@ def random_prompt(`
`261`	`261`	`assert len(examples) == num_fewshots`
`262`	`262`
`263`	`263`	`body = "\n\n".join(`
`264`		`- f"## Example {idx + 1}\n{example.prompt(mode, index=idx + 1)}"`
	`264`	`+ f"## Example {idx + 1}\n{example.prompt(mode, index=idx + 1 if LLAMA3 and mode == 'I->R' else None)}"`
`265`	`265`	`for idx, example in enumerate(examples)`
`266`	`266`	`)`
`267`	`267`	`# content = f"{self.system_prompt}\n\n{body}"`