Spaces:
Runtime error
Runtime error
parokshsaxena
commited on
Commit
•
b79897e
1
Parent(s):
0bd8c50
using zero GPU after moving repo to org
Browse files- gradio_demo/app.py +2 -0
- requirements_bk.txt +18 -17
gradio_demo/app.py
CHANGED
@@ -1,5 +1,6 @@
|
|
1 |
import sys
|
2 |
sys.path.append('./')
|
|
|
3 |
from PIL import Image
|
4 |
import gradio as gr
|
5 |
from src.tryon_pipeline import StableDiffusionXLInpaintPipeline as TryonPipeline
|
@@ -123,6 +124,7 @@ pipe = TryonPipeline.from_pretrained(
|
|
123 |
)
|
124 |
pipe.unet_encoder = UNet_Encoder
|
125 |
|
|
|
126 |
def start_tryon(dict,garm_img,garment_des,is_checked,is_checked_crop,denoise_steps,seed):
|
127 |
|
128 |
openpose_model.preprocessor.body_estimation.model.to(device)
|
|
|
1 |
import sys
|
2 |
sys.path.append('./')
|
3 |
+
import spaces
|
4 |
from PIL import Image
|
5 |
import gradio as gr
|
6 |
from src.tryon_pipeline import StableDiffusionXLInpaintPipeline as TryonPipeline
|
|
|
124 |
)
|
125 |
pipe.unet_encoder = UNet_Encoder
|
126 |
|
127 |
+
@spaces.GPU
|
128 |
def start_tryon(dict,garm_img,garment_des,is_checked,is_checked_crop,denoise_steps,seed):
|
129 |
|
130 |
openpose_model.preprocessor.body_estimation.model.to(device)
|
requirements_bk.txt
CHANGED
@@ -1,23 +1,24 @@
|
|
1 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
2 |
torch==2.0.1
|
3 |
-
|
4 |
-
|
5 |
-
numpy==1.24.4
|
6 |
-
scipy==1.10.1
|
7 |
-
scikit-image==0.21.0
|
8 |
-
opencv-python==4.7.0.72
|
9 |
-
pillow==9.4.0
|
10 |
-
diffusers==0.25.0
|
11 |
transformers==4.36.2
|
12 |
-
|
13 |
-
matplotlib==3.7.4
|
14 |
-
tqdm==4.64.1
|
15 |
-
config==0.5.1
|
16 |
einops==0.7.0
|
17 |
-
|
18 |
-
|
19 |
-
|
|
|
20 |
fvcore
|
21 |
cloudpickle
|
22 |
omegaconf
|
23 |
-
pycocotools
|
|
|
|
|
|
|
|
1 |
+
#pytorch==2.1.0
|
2 |
+
#pytorch-cuda==11.8
|
3 |
+
#torchaudio=2.0.2
|
4 |
+
#torchtriton=2.0.0
|
5 |
+
#torchvision=0.15.2
|
6 |
+
|
7 |
+
accelerate==0.25.0
|
8 |
torch==2.0.1
|
9 |
+
torchmetrics==1.2.1
|
10 |
+
tqdm==4.66.1
|
|
|
|
|
|
|
|
|
|
|
|
|
11 |
transformers==4.36.2
|
12 |
+
diffusers==0.25.0
|
|
|
|
|
|
|
13 |
einops==0.7.0
|
14 |
+
bitsandbytes==0.39.0
|
15 |
+
scipy==1.11.1
|
16 |
+
opencv-python
|
17 |
+
gradio==4.36.1
|
18 |
fvcore
|
19 |
cloudpickle
|
20 |
omegaconf
|
21 |
+
pycocotools
|
22 |
+
basicsr
|
23 |
+
av
|
24 |
+
onnxruntime==1.16.2
|