22

README.md

@@ -12,7 +12,8 @@ short_description: Supports Canny, HED, Depth, Pose and MLSD
 models:
     - Tongyi-MAI/Z-Image-Turbo
     - alibaba-pai/Z-Image-Turbo-Fun-Controlnet-Union
-app_build_command: echo "Starting Space..."
+preload_from_hub:
+    - alibaba-pai/Z-Image-Turbo-Fun-Controlnet-Union
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -57,7 +57,11 @@ text_encoder = Qwen3ForCausalLM.from_pretrained(
     MODEL_LOCAL, subfolder="text_encoder", torch_dtype=weight_dtype,
     low_cpu_mem_usage=False,
 )
-scheduler = FlowMatchEulerDiscreteScheduler(num_train_timesteps=1000, shift=3)
+# scheduler = FlowMatchEulerDiscreteScheduler(num_train_timesteps=1000, shift=3)
+scheduler = FlowMatchEulerDiscreteScheduler.from_pretrained(
+    MODEL_LOCAL, 
+    subfolder="scheduler"
+)
 pipe = ZImageControlPipeline(
     vae=vae,
     tokenizer=tokenizer,