Upload train_smol_discharge.py with huggingface_hub

train_smol_discharge.py

@@ -9,6 +9,15 @@
 # ]
 # ///
 
+import os
+from huggingface_hub import login
+
+# Login with token from environment (set via secrets)
+token = os.environ.get("HF_TOKEN")
+if token:
+    login(token=token)
+    print("Logged in to HuggingFace Hub")
+
 from datasets import load_dataset
 from peft import LoraConfig
 from transformers import AutoTokenizer, AutoModelForCausalLM