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
|