fix

scripts/train.py

@@ -52,12 +52,12 @@ def main():
                     formatted += f"{role.capitalize()}: {content}\n"
             texts.append(formatted)
     
-    return tokenizer(
-        texts,
-        truncation=True,
-        max_length=4096,
-        padding="max_length"
-    )
+        return tokenizer(
+            texts,
+            truncation=True,
+            max_length=4096,
+            padding="max_length"
+        )
 
     
     tokenized_dataset = dataset.map(