Upload train_flux_lora_v9.py with huggingface_hub

train_flux_lora_v9.py

@@ -0,0 +1,53 @@
+# /// script
+# requires-python = ">=3.10"
+# dependencies = [
+#     "torch>=2.0.0",
+#     "diffusers>=0.25.0",
+#     "transformers>=4.35.0",
+#     "accelerate>=0.24.0",
+#     "peft>=0.7.0",
+#     "huggingface-hub>=0.20.0",
+#     "safetensors>=0.4.0",
+#     "Pillow>=10.0.0",
+#     "numpy>=1.24.0",
+#     "tqdm>=4.66.0",
+# ]
+# ///
+
+import os
+import sys
+print("="*70)
+print("FLUX LoRA Training v9 - Token Fix")
+print("="*70)
+
+# Get token - try multiple sources
+token = os.environ.get("HF_TOKEN")
+if not token or token == "$HF_TOKEN":
+    # Try to read from file if env var not set properly
+    try:
+        with open("/secrets/HF_TOKEN") as f:
+            token = f.read().strip()
+    except:
+        pass
+
+if not token or token == "$HF_TOKEN":
+    print("ERROR: HF_TOKEN not set correctly")
+    print(f"  Value: {token}")
+    print("  Env vars:", list(os.environ.keys()))
+    sys.exit(1)
+
+print(f"Token loaded: {token[:10]}...")
+
+# Set token for huggingface_hub
+os.environ["HF_TOKEN"] = token
+os.environ["HUGGINGFACE_TOKEN"] = token
+
+# Now import and login
+from huggingface_hub import login, HfApi
+login(token=token, add_to_git_credential=False)
+print("✅ Login successful")
+
+from diffusers import FluxPipeline
+print("✅ FluxPipeline imported")
+
+print("\n✅ SUCCESS: All imports work!")