cocktailpeanut commited on
Commit
44a33d6
1 Parent(s): 747b988
Files changed (2) hide show
  1. app.py +2 -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
- @spaces.GPU(duration=80)
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