techfreakworm commited on
Commit
6bd8e31
·
unverified ·
1 Parent(s): 0e7a9cf

perf(preload): broaden globs to capture configs + index.jsons (~47 GB total)

Browse files

Old preload listed only *.safetensors and tokenizer/*, which missed:
- text_encoder/config.json + model.safetensors.index.json (shard map)
- transformer/config.json + .index.json
- vae/config.json
- scheduler/scheduler_config.json
- model_index.json (diffusers top-level)

Result: every cold-start inference fired 5-8 small hf_hub_download calls
to fill those gaps before the pipeline could load. Annoying first-call
latency on the Space.

Broaden to transformer/* etc. so HF preload pulls each subdir wholesale.
Skip the shared encoder/vae/tokenizer from Z-Image-Turbo (identical to
Base's). Keep alibaba-pai and lllyasviel/Annotators tight — their repos
contain other models we don't want.

Sizes (from HF API):
Tongyi-MAI/Z-Image: ~20.6 GB
Tongyi-MAI/Z-Image-Turbo: ~24.6 GB (transformer subdir only)
alibaba-pai/...Union-2.1: ~2 GB (one file)
lllyasviel/Annotators: 67 MB (one file)
──────────────────────────────────
Total: ~47 GB (cap is 150 GB)

README.md CHANGED
@@ -10,8 +10,8 @@ python_version: "3.11"
10
  suggested_hardware: zero-a10g
11
  hf_oauth: false
12
  preload_from_hub:
13
- - Tongyi-MAI/Z-Image transformer/diffusion_pytorch_model.safetensors,text_encoder/*.safetensors,vae/diffusion_pytorch_model.safetensors,tokenizer/*
14
- - Tongyi-MAI/Z-Image-Turbo transformer/diffusion_pytorch_model.safetensors
15
  - alibaba-pai/Z-Image-Turbo-Fun-Controlnet-Union-2.1 Z-Image-Turbo-Fun-Controlnet-Union-2.1-8steps.safetensors
16
  - lllyasviel/Annotators RealESRGAN_x4plus.pth
17
  ---
 
10
  suggested_hardware: zero-a10g
11
  hf_oauth: false
12
  preload_from_hub:
13
+ - Tongyi-MAI/Z-Image transformer/*,text_encoder/*,vae/*,tokenizer/*,scheduler/*,model_index.json
14
+ - Tongyi-MAI/Z-Image-Turbo transformer/*,scheduler/*,model_index.json
15
  - alibaba-pai/Z-Image-Turbo-Fun-Controlnet-Union-2.1 Z-Image-Turbo-Fun-Controlnet-Union-2.1-8steps.safetensors
16
  - lllyasviel/Annotators RealESRGAN_x4plus.pth
17
  ---
docs/superpowers/specs/2026-05-13-z-image-studio-design.md CHANGED
@@ -333,8 +333,8 @@ python_version: "3.11"
333
  suggested_hardware: zero-a10g
334
  hf_oauth: false
335
  preload_from_hub:
336
- - Tongyi-MAI/Z-Image transformer/diffusion_pytorch_model.safetensors,text_encoder/*.safetensors,vae/diffusion_pytorch_model.safetensors,tokenizer/*
337
- - Tongyi-MAI/Z-Image-Turbo transformer/diffusion_pytorch_model.safetensors
338
  - alibaba-pai/Z-Image-Turbo-Fun-Controlnet-Union-2.1 Z-Image-Turbo-Fun-Controlnet-Union-2.1-8steps.safetensors
339
  - lllyasviel/Annotators RealESRGAN_x4plus.pth
340
  ---
 
333
  suggested_hardware: zero-a10g
334
  hf_oauth: false
335
  preload_from_hub:
336
+ - Tongyi-MAI/Z-Image transformer/*,text_encoder/*,vae/*,tokenizer/*,scheduler/*,model_index.json
337
+ - Tongyi-MAI/Z-Image-Turbo transformer/*,scheduler/*,model_index.json
338
  - alibaba-pai/Z-Image-Turbo-Fun-Controlnet-Union-2.1 Z-Image-Turbo-Fun-Controlnet-Union-2.1-8steps.safetensors
339
  - lllyasviel/Annotators RealESRGAN_x4plus.pth
340
  ---