Upload folder using huggingface_hub

test.py

@@ -1,8 +1,9 @@
-from qwen_prompt_expander import QwenPromptExpander
+from diffusers.modular_pipelines import ModularPipelineBlocks
 
-# expander = QwenPromptExpander().init_pipeline()
-# output = expander(prompt="a dog sitting by the river, watching the sunset")
-# print(f"{output.values['prompt']=}")
-
-
-QwenPromptExpander().save_pretrained(".")
+expander_qwen = ModularPipelineBlocks.from_pretrained(
+    "diffusers-internal-dev/qwen-prompt-expander",
+    trust_remote_code=True,
+)
+expander_qwen = expander_qwen.init_pipeline()
+output = expander_qwen(prompt="a dog sitting by the river, watching the sunset")
+print(f"{output.values['prompt']=}")