Create download_flux.py

download_flux.py

@@ -0,0 +1,15 @@
+#!/usr/bin/env python3
+import os
+from huggingface_hub import hf_hub_download
+
+token = open('/run/secrets/HF_TOKEN').read().strip()
+for subdir, filename in [['models/unet', 'flux1-schnell.safetensors'],
+                         ['models/vae', 'ae.safetensors']]:
+    os.makedirs(subdir, exist_ok=True)
+    hf_hub_download(
+        repo_id="black-forest-labs/FLUX.1-schnell",
+        filename=filename,
+        token=token,
+        local_dir=subdir,
+        force_download=True
+    )