Hugging Face's logo

Spaces:
levihsu
/
OOTDiffusion
Running on Zero

App Files Community
20
levihsu commited on
Commit
7ab0607
•
1 Parent(s): 83ed56c

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 = "levihsu/OOTDiffusion/checkpoints/ootd"
29
+ UNET_PATH = "levihsu/OOTDiffusion/checkpoints/ootd/ootd_hd/checkpoint-36000"
30
+ MODEL_PATH = "levihsu/OOTDiffusion/checkpoints/ootd"
31
 
32
  class OOTDiffusionHD:
33