moose Claude Opus 4.7 (1M context) commited on
Commit
3e914c4
·
1 Parent(s): e64cecc

Disable AOTI compile to dodge spaces==0.49.3 NVML regression

Browse files

HF's build pipeline force-installs spaces==0.49.3 today, which has a
regression in zero.torch.patching._move() that NVML-asserts during
worker_init before the AOTI compile body runs. No way to override the
pin from inside requirements.txt. Skipping the optimize_pipeline_ call
gets the space booting on FA3 + Phr00t v18 alone. Restore once HF
propagates spaces==0.50.0 to the build pipeline.

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>

Files changed (2) hide show
  1. app.py +4 -1
  2. requirements.txt +0 -1
app.py CHANGED
@@ -123,7 +123,10 @@ pipe.transformer.set_attn_processor(QwenDoubleStreamAttnProcessorFA3())
123
 
124
  # --- Ahead-of-time compilation ---
125
  # Note: optimize_pipeline_ handles text encoder offloading internally to save memory during torch.export
126
- optimize_pipeline_(pipe, image=[Image.new("RGB", (1024, 1024)), Image.new("RGB", (1024, 1024))], prompt="prompt")
 
 
 
127
 
128
  # --- UI Constants and Helpers ---
129
  MAX_SEED = np.iinfo(np.int32).max
 
123
 
124
  # --- Ahead-of-time compilation ---
125
  # Note: optimize_pipeline_ handles text encoder offloading internally to save memory during torch.export
126
+ # DISABLED 2026-05-12: HF build pipeline force-pins spaces==0.49.3 which has a regression in
127
+ # zero.torch.patching._move() — NVML assert during worker_init kills AOTI compile at startup.
128
+ # Restore once HF bumps the pipeline to spaces==0.50.0+.
129
+ # optimize_pipeline_(pipe, image=[Image.new("RGB", (1024, 1024)), Image.new("RGB", (1024, 1024))], prompt="prompt")
130
 
131
  # --- UI Constants and Helpers ---
132
  MAX_SEED = np.iinfo(np.int32).max
requirements.txt CHANGED
@@ -1,6 +1,5 @@
1
  git+https://github.com/huggingface/diffusers.git
2
 
3
- spaces==0.50.0
4
  transformers
5
  accelerate
6
  safetensors
 
1
  git+https://github.com/huggingface/diffusers.git
2
 
 
3
  transformers
4
  accelerate
5
  safetensors