File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -1149,7 +1149,7 @@ def call(self, inputs):
11491149
11501150# Process ALL original samples from data - REAL WORLD USAGE
11511151generated_texts = []
1152- for i , original_text in enumerate (data [:5 ]): # Process first 5 samples
1152+ for i , original_text in enumerate (train_samples_list_text [:5 ]): # Process first 5 samples
11531153 print (f"\n Processing sample { i + 1 } ..." )
11541154
11551155 # Extract prompt part (everything up to and including </prompt>)
@@ -1217,7 +1217,7 @@ def call(self, inputs):
12171217print ("=" * 50 )
12181218
12191219generated_texts_all = []
1220- for i , text in enumerate (data ):
1220+ for i , text in enumerate (train_samples_list_text ):
12211221 # Extract prompt part (everything up to and including </prompt>)
12221222 if '</prompt>' in text :
12231223 prompt_text = text .split ('</prompt>' )[0 ] + '</prompt>'
@@ -1260,7 +1260,7 @@ def call(self, inputs):
12601260print ("=" * 50 )
12611261
12621262# generated_texts_all = []
1263- # for i, text in enumerate(data [:3]): # Process first 3 for demo
1263+ # for i, text in enumerate(train_samples_list_text [:3]): # Process first 3 for demo
12641264# # Split such that everything before </prompt> or the entire text if </prompt> is not present
12651265# if '</prompt>' in text:
12661266# prompt_text = text.split('</prompt>')[0] + '</prompt>'
You can’t perform that action at this time.
0 commit comments