Spaces:
Runtime error
Runtime error
cocktailpeanut
commited on
Commit
•
44a33d6
1
Parent(s):
747b988
localize
Browse files- app.py +2 -2
- requirements.txt +3 -3
app.py
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
import os
|
2 |
from huggingface_hub import snapshot_download
|
3 |
-
import spaces
|
4 |
import torch
|
5 |
torch.jit.script = lambda f: f
|
6 |
os.makedirs("/home/user/app/ckpts", exist_ok=True)
|
@@ -34,7 +34,7 @@ def get_strings(lang):
|
|
34 |
args, gen, enhancer = inferencer()
|
35 |
strings = get_strings("en")
|
36 |
|
37 |
-
|
38 |
def infer(
|
39 |
prompt,
|
40 |
negative_prompt,
|
|
|
1 |
import os
|
2 |
from huggingface_hub import snapshot_download
|
3 |
+
#import spaces
|
4 |
import torch
|
5 |
torch.jit.script = lambda f: f
|
6 |
os.makedirs("/home/user/app/ckpts", exist_ok=True)
|
|
|
34 |
args, gen, enhancer = inferencer()
|
35 |
strings = get_strings("en")
|
36 |
|
37 |
+
#@spaces.GPU(duration=80)
|
38 |
def infer(
|
39 |
prompt,
|
40 |
negative_prompt,
|
requirements.txt
CHANGED
@@ -3,13 +3,13 @@ timm==0.9.16
|
|
3 |
diffusers==0.21.2
|
4 |
peft==0.10.0
|
5 |
protobuf==3.19.0
|
6 |
-
torchvision==0.17.0
|
7 |
transformers==4.37.2
|
8 |
accelerate==0.29.3
|
9 |
loguru==0.7.2
|
10 |
einops==0.7.0
|
11 |
sentencepiece==0.1.99
|
12 |
-
cuda-python==11.7.1
|
13 |
onnxruntime==1.12.1
|
14 |
onnx==1.12.0
|
15 |
nvidia-pyindex==1.0.9
|
@@ -17,4 +17,4 @@ onnx-graphsurgeon==0.3.27
|
|
17 |
polygraphy==0.47.1
|
18 |
pandas==2.0.3
|
19 |
gradio==4.31.0
|
20 |
-
torch==2.2.0
|
|
|
3 |
diffusers==0.21.2
|
4 |
peft==0.10.0
|
5 |
protobuf==3.19.0
|
6 |
+
#torchvision==0.17.0
|
7 |
transformers==4.37.2
|
8 |
accelerate==0.29.3
|
9 |
loguru==0.7.2
|
10 |
einops==0.7.0
|
11 |
sentencepiece==0.1.99
|
12 |
+
#cuda-python==11.7.1
|
13 |
onnxruntime==1.12.1
|
14 |
onnx==1.12.0
|
15 |
nvidia-pyindex==1.0.9
|
|
|
17 |
polygraphy==0.47.1
|
18 |
pandas==2.0.3
|
19 |
gradio==4.31.0
|
20 |
+
#torch==2.2.0
|