levihsu commited on
Commit
7ea6f18
1 Parent(s): 445bb6f

Update ootd/inference_ootd.py

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