Spaces:

levihsu
/

OOTDiffusion

Running on Zero

levihsu commited on Mar 8

Commit

71b73b9

•

1 Parent(s): d86f085

Update ootd/inference_ootd_dc.py

Files changed (1) hide show

ootd/inference_ootd_dc.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_dc/checkpoint-36000"
-MODEL_PATH = "../checkpoints/ootd"
 class OOTDiffusionDC:

 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_dc/checkpoint-36000"
+MODEL_PATH = "./checkpoints/ootd"
 class OOTDiffusionDC: