Upload train_survival.py with huggingface_hub

train_survival.py

@@ -10,6 +10,7 @@ import torch
 import os
 
 # Configuration
+# Switching to Llama 3.2 3B Instruct to avoid BFloat16 issues
 MODEL_ID = "Qwen/Qwen2.5-3B-Instruct"
 DATASET_ID = "sunkencity/survival-instruct"
 OUTPUT_MODEL_ID = "sunkencity/survival-expert-3b"
@@ -95,4 +96,4 @@ trainer.train()
 
 print("Pushing to hub...")
 trainer.push_to_hub()
-print("Done!")
+print("Done!")