Spaces:
Running
on
Zero
Running
on
Zero
torch 2.8 + torchao + spaces upgrade
Browse files- app.py +5 -0
- 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
|