levihsu commited on
Commit
d86f085
β€’
1 Parent(s): 4c4a8aa

Update ootd/inference_ootd_hd.py

Browse files
Files changed (1) hide show
  1. ootd/inference_ootd_hd.py +4 -4
ootd/inference_ootd_hd.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_hd/checkpoint-36000"
30
- MODEL_PATH = "../checkpoints/ootd"
31
 
32
  class OOTDiffusionHD:
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_hd/checkpoint-36000"
30
+ MODEL_PATH = "./checkpoints/ootd"
31
 
32
  class OOTDiffusionHD:
33