Update ootd/inference_ootd.py
Browse files- 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 = "
|
| 29 |
-
VAE_PATH = "
|
| 30 |
-
UNET_PATH = "
|
| 31 |
-
MODEL_PATH = "
|
| 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 = "./checkpoints/ootd"
|
| 30 |
+
UNET_PATH = "./checkpoints/ootd/ootd_hd/checkpoint-36000"
|
| 31 |
+
MODEL_PATH = "./checkpoints/ootd"
|
| 32 |
|
| 33 |
class OOTDiffusion:
|
| 34 |
|