Upload train_survival.py with huggingface_hub

train_survival.py

@@ -55,7 +55,8 @@ training_args = SFTConfig(
     hub_model_id=OUTPUT_MODEL_ID,
     fp16=True,
     dataset_text_field="text",
-    packing=False
+    packing=False,
+    max_length=1024 # Correct parameter name for SFTConfig
 )
 
 def formatting_prompts_func(example):
@@ -74,8 +75,7 @@ trainer = SFTTrainer(
     peft_config=peft_config,
     formatting_func=formatting_prompts_func,
     args=training_args,
-    processing_class=tokenizer, # New name for tokenizer
-    max_seq_length=1024 # Passed here
+    processing_class=tokenizer,
 )
 
 print("Starting training...")
@@ -83,4 +83,4 @@ trainer.train()
 
 print("Pushing to hub...")
 trainer.push_to_hub()
-print("Done!")
+print("Done!")