parokshsaxena commited on
Commit
b79897e
1 Parent(s): 0bd8c50

using zero GPU after moving repo to org

Browse files
Files changed (2) hide show
  1. gradio_demo/app.py +2 -0
  2. 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
- transformers==4.36.2
 
 
 
 
 
 
2
  torch==2.0.1
3
- torchvision==0.15.2
4
- torchaudio==2.0.2
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
- accelerate==0.26.1
13
- matplotlib==3.7.4
14
- tqdm==4.64.1
15
- config==0.5.1
16
  einops==0.7.0
17
- onnxruntime==1.16.2
18
- basicsr
19
- av
 
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