levihsu commited on
Commit
71b73b9
1 Parent(s): d86f085

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 = "./checkpoints/ootd"
29
+ UNET_PATH = "./checkpoints/ootd/ootd_dc/checkpoint-36000"
30
+ MODEL_PATH = "./checkpoints/ootd"
31
 
32
  class OOTDiffusionDC:
33