cbensimon HF Staff commited on
Commit
43846b1
·
1 Parent(s): 288103a

torch 2.8 + torchao + spaces upgrade

Browse files
Files changed (2) hide show
  1. app.py +5 -0
  2. requirements.txt +2 -1
app.py CHANGED
@@ -1,3 +1,8 @@
 
 
 
 
 
1
  import gradio as gr
2
  import numpy as np
3
  import spaces
 
1
+ # PyTorch 2.8 (temporary hack)
2
+ import os
3
+ os.system('pip install --upgrade --pre --extra-index-url https://download.pytorch.org/whl/nightly/cu126 "torch<2.9" spaces')
4
+
5
+ # Actual demo code
6
  import gradio as gr
7
  import numpy as np
8
  import spaces
requirements.txt CHANGED
@@ -2,4 +2,5 @@ transformers
2
  git+https://github.com/huggingface/diffusers.git
3
  accelerate
4
  safetensors
5
- sentencepiece
 
 
2
  git+https://github.com/huggingface/diffusers.git
3
  accelerate
4
  safetensors
5
+ sentencepiece
6
+ torchao