Hugging Face's logo

Spaces:
levihsu
/
OOTDiffusion
Running on Zero

App Files Community
32
levihsu commited on
Commit
445bb6f
·
verified ·
1 Parent(s): 7ab0607

Update ootd/inference_ootd_dc.py

Browse files
Files changed (1) hide show
  1. ootd/inference_ootd_dc.py +4 -4
ootd/inference_ootd_dc.py CHANGED
@@ -24,10 +24,10 @@ import torch.nn.functional as F
24
  from transformers import AutoProcessor, CLIPVisionModelWithProjection
25
  from transformers import CLIPTextModel, CLIPTokenizer
26
 
27
- VIT_PATH = "../checkpoints/clip-vit-large-patch14"
28
- VAE_PATH = "../checkpoints/ootd"
29
- UNET_PATH = "../checkpoints/ootd/ootd_dc/checkpoint-36000"
30
- MODEL_PATH = "../checkpoints/ootd"
31
 
32
  class OOTDiffusionDC:
33
 
 
24
  from transformers import AutoProcessor, CLIPVisionModelWithProjection
25
  from transformers import CLIPTextModel, CLIPTokenizer
26
 
27
+ VIT_PATH = "openai/clip-vit-large-patch14"
28
+ VAE_PATH = "levihsu/OOTDiffusion/checkpoints/ootd"
29
+ UNET_PATH = "levihsu/OOTDiffusion/checkpoints/ootd/ootd_dc/checkpoint-36000"
30
+ MODEL_PATH = "levihsu/OOTDiffusion/checkpoints/ootd"
31
 
32
  class OOTDiffusionDC:
33