Upload folder using huggingface_hub

compose/merge.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import os
 from pathlib import Path
 from typing import Any
 
@@ -30,10 +31,11 @@ def _load_pipe():
     from diffusers import Flux2KleinPipeline
 
     device, dtype = _device()
+    token = os.environ.get("HF_TOKEN")
     pipe = Flux2KleinPipeline.from_pretrained(
         BASE_INFERENCE_MODEL,
         torch_dtype=dtype,
-        token=None,
+        token=token,
     )
     pipe = pipe.to(device)
     _PIPE = pipe
@@ -45,7 +47,7 @@ def _load_adapters(pipe, lora_ids: list[str], registry_by_id: dict[str, dict[str
         if lora_id in _LOADED_ADAPTERS:
             continue
         repo = registry_by_id[lora_id]["hf_repo"]
-        pipe.load_lora_weights(repo, adapter_name=lora_id)
+        pipe.load_lora_weights(repo, adapter_name=lora_id, token=os.environ.get("HF_TOKEN"))
         _LOADED_ADAPTERS.add(lora_id)
 
 

requirements.txt

@@ -3,7 +3,7 @@ pydantic>=2.0
 PyYAML>=6.0
 Pillow>=10.0
 requests>=2.31
-huggingface_hub>=0.26
+huggingface_hub>=0.26,<1.0
 peft>=0.14
 accelerate>=1.0
 torch