Spaces:
Running
on
Zero
Running
on
Zero
ohayonguy
commited on
Commit
·
ce63530
1
Parent(s):
88b7196
test
Browse files- app.py +8 -0
- requirements.txt +1 -2
app.py
CHANGED
|
@@ -1,3 +1,11 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
import os
|
| 2 |
if os.getenv('SPACES_ZERO_GPU') == "true":
|
| 3 |
os.environ['SPACES_ZERO_GPU'] = "1"
|
|
|
|
| 1 |
+
import subprocess
|
| 2 |
+
import sys
|
| 3 |
+
|
| 4 |
+
def install(package):
|
| 5 |
+
subprocess.check_call([sys.executable, "-m", "pip", "install", package])
|
| 6 |
+
|
| 7 |
+
install('natten==0.17.1')
|
| 8 |
+
|
| 9 |
import os
|
| 10 |
if os.getenv('SPACES_ZERO_GPU') == "true":
|
| 11 |
os.environ['SPACES_ZERO_GPU'] = "1"
|
requirements.txt
CHANGED
|
@@ -18,5 +18,4 @@ huggingface_hub==0.24.6
|
|
| 18 |
wandb
|
| 19 |
timm
|
| 20 |
torchmetrics
|
| 21 |
-
torch-fidelity==0.3.0
|
| 22 |
-
natten==0.17.1
|
|
|
|
| 18 |
wandb
|
| 19 |
timm
|
| 20 |
torchmetrics
|
| 21 |
+
torch-fidelity==0.3.0
|
|
|