OOTDiffusion

Build error

levihsu commited on Mar 8

Commit

d86f085

•

1 Parent(s): 4c4a8aa

Update ootd/inference_ootd_hd.py

Files changed (1) hide show

ootd/inference_ootd_hd.py CHANGED Viewed

@@ -24,10 +24,10 @@ import torch.nn.functional as F
 from transformers import AutoProcessor, CLIPVisionModelWithProjection
 from transformers import CLIPTextModel, CLIPTokenizer
-VIT_PATH = "../checkpoints/clip-vit-large-patch14"
-VAE_PATH = "../checkpoints/ootd"
-UNET_PATH = "../checkpoints/ootd/ootd_hd/checkpoint-36000"
-MODEL_PATH = "../checkpoints/ootd"
 class OOTDiffusionHD:

 from transformers import AutoProcessor, CLIPVisionModelWithProjection
 from transformers import CLIPTextModel, CLIPTokenizer
+VIT_PATH = "openai/clip-vit-large-patch14"
+VAE_PATH = "./checkpoints/ootd"
+UNET_PATH = "./checkpoints/ootd/ootd_hd/checkpoint-36000"
+MODEL_PATH = "./checkpoints/ootd"
 class OOTDiffusionHD: