Hugging Face's logo

Spaces:
zhengchong
/
CatVTON
Running on Zero

App Files Community
3
zhengchong commited on
Commit
6927e60
·
1 Parent(s): 7b80416

chore: Update app.py to fix torch.jit.script assignment

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ import torch
11
  from diffusers.image_processor import VaeImageProcessor
12
  from huggingface_hub import snapshot_download
13
  from PIL import Image
14
-
15
  from model.cloth_masker import AutoMasker, vis_mask
16
  from model.pipeline import CatVTONPipeline
17
  from utils import init_weight_dtype, resize_and_crop, resize_and_padding
 
11
  from diffusers.image_processor import VaeImageProcessor
12
  from huggingface_hub import snapshot_download
13
  from PIL import Image
14
+ torch.jit.script = lambda f: f
15
  from model.cloth_masker import AutoMasker, vis_mask
16
  from model.pipeline import CatVTONPipeline
17
  from utils import init_weight_dtype, resize_and_crop, resize_and_padding