Upload folder using huggingface_hub

synthcxr/pipeline.py

@@ -38,24 +38,24 @@ def load_pipeline(
     model_configs = [
         ModelConfig(
             model_id=model_id,
-            origin_file_pattern="transformer/diffusion_pytorch_model*.safetensors",
+            origin_file_pattern="Qwen-Image-Edit-2511/transformer/diffusion_pytorch_model*.safetensors",
         ),
         ModelConfig(
             model_id=TEXT_ENCODER_MODEL_ID,
-            origin_file_pattern="text_encoder/model*.safetensors",
+            origin_file_pattern="Qwen-Image/text_encoder/model*.safetensors",
         ),
         ModelConfig(
             model_id=TEXT_ENCODER_MODEL_ID,
-            origin_file_pattern="vae/diffusion_pytorch_model.safetensors",
+            origin_file_pattern="Qwen-Image/vae/diffusion_pytorch_model.safetensors",
         ),
     ]
     tokenizer_config = ModelConfig(
         model_id=TEXT_ENCODER_MODEL_ID,
-        origin_file_pattern="tokenizer/",
+        origin_file_pattern="Qwen-Image/tokenizer/",
     )
     processor_config = ModelConfig(
         model_id=PROCESSOR_MODEL_ID,
-        origin_file_pattern="processor/",
+        origin_file_pattern="Qwen-Image-Edit/processor/",
     )
     pipe = QwenImagePipeline.from_pretrained(
         torch_dtype=torch_dtype,