Weijie Lyu Claude Opus 4.8 (1M context) commited on
Commit
4f828ee
·
1 Parent(s): d7cc8a7

Move torch stack to 2.11.0 to resolve xformers conflict

Browse files

xformers 0.0.33 hard-pins torch==2.9.0 (ZeroGPU-unsupported), which
conflicted with torchvision 0.24.1 (torch==2.9.1) and broke pip
resolution. xformers 0.0.35 relaxed to torch>=2.10, so move the whole
trio to the agreeing, ZeroGPU-supported line:
torch 2.11.0, torchvision 0.26.0, xformers 0.0.35
Update the startup cu130 reinstall to the same versions.

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

Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +8 -6
app.py CHANGED
@@ -52,7 +52,7 @@ def _ensure_cu130_torch():
52
  _sys.executable, "-m", "pip", "install",
53
  "--no-cache-dir", "--force-reinstall", "--no-deps",
54
  "--index-url", "https://download.pytorch.org/whl/cu130",
55
- "torch==2.9.1+cu130", "torchvision==0.24.1+cu130", "xformers==0.0.33",
56
  ])
57
  print("[CU130-CHECK] cu130 stack installed")
58
 
 
52
  _sys.executable, "-m", "pip", "install",
53
  "--no-cache-dir", "--force-reinstall", "--no-deps",
54
  "--index-url", "https://download.pytorch.org/whl/cu130",
55
+ "torch==2.11.0+cu130", "torchvision==0.26.0+cu130", "xformers==0.0.35",
56
  ])
57
  print("[CU130-CHECK] cu130 stack installed")
58
 
requirements.txt CHANGED
@@ -1,9 +1,11 @@
1
- # torch must be 2.9.1 (ZeroGPU-supported, bare version for the config validator).
 
 
2
  # The CUDA build matters: the ZeroGPU container ships CUDA 13.0, so torch must be
3
- # cu130 for diff-gaussian-rasterization to compile. Build-time pip kept resolving
4
- # the cu128 wheel here, so the cu130 reinstall is forced at startup in app.py.
5
- torch==2.9.1
6
- torchvision==0.24.1
7
  numpy==1.26.4
8
  pillow==10.4.0
9
  diffusers[torch]==0.30.3
@@ -23,7 +25,7 @@ videoio==0.3.0
23
  gradio==5.49.1
24
  matplotlib==3.7.5
25
  onnxruntime>=1.15.0
26
- xformers==0.0.33
27
  scikit-image==0.21.0
28
  jaxtyping==0.2.19
29
  pytorch-msssim==1.0.0
 
1
+ # torch 2.11.0 is ZeroGPU-supported (bare version for the config validator) and is
2
+ # the version line where torchvision 0.26.0 and xformers 0.0.35 agree (xformers
3
+ # 0.0.33 hard-pinned torch==2.9.0, which ZeroGPU rejects; 0.0.35 relaxed to >=2.10).
4
  # The CUDA build matters: the ZeroGPU container ships CUDA 13.0, so torch must be
5
+ # cu130 for diff-gaussian-rasterization to compile. Build-time pip resolves the
6
+ # cu128 wheel here, so the cu130 reinstall is forced at startup in app.py.
7
+ torch==2.11.0
8
+ torchvision==0.26.0
9
  numpy==1.26.4
10
  pillow==10.4.0
11
  diffusers[torch]==0.30.3
 
25
  gradio==5.49.1
26
  matplotlib==3.7.5
27
  onnxruntime>=1.15.0
28
+ xformers==0.0.35
29
  scikit-image==0.21.0
30
  jaxtyping==0.2.19
31
  pytorch-msssim==1.0.0