xinjjj commited on
Commit
db38ead
·
1 Parent(s): 41cb729

fix(texture): use cu126 torch with sm120 wheels

Browse files
Files changed (2) hide show
  1. app.py +0 -2
  2. requirements.txt +1 -1
app.py CHANGED
@@ -17,8 +17,6 @@
17
 
18
  import os
19
 
20
- # ZeroGPU/cu128 can hit an NVML assert inside PyTorch CUDACachingAllocator.
21
- os.environ.setdefault("PYTORCH_NO_CUDA_MEMORY_CACHING", "1")
22
  os.environ["GRADIO_APP"] = "texture_edit"
23
  import gradio as gr
24
  from app_style import custom_theme, image_css, lighting_css
 
17
 
18
  import os
19
 
 
 
20
  os.environ["GRADIO_APP"] = "texture_edit"
21
  import gradio as gr
22
  from app_style import custom_theme, image_css, lighting_css
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
- --extra-index-url https://download.pytorch.org/whl/cu128
2
 
3
  setuptools==80.10.2
4
  wheel
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu126
2
 
3
  setuptools==80.10.2
4
  wheel