Spaces:
Paused
Paused
fixed import
Browse files- api/model/src/eval.py +1 -1
api/model/src/eval.py
CHANGED
|
@@ -14,7 +14,7 @@ from diffusers import UNet2DConditionModel
|
|
| 14 |
from transformers import CLIPTextModel, CLIPTokenizer
|
| 15 |
|
| 16 |
# custom imports
|
| 17 |
-
from model.src.datasets.
|
| 18 |
from model.src.datasets.vitonhd import VitonHDDataset
|
| 19 |
from model.src.mgd_pipelines.mgd_pipe import MGDPipe
|
| 20 |
from model.src.mgd_pipelines.mgd_pipe_disentangled import MGDPipeDisentangled
|
|
|
|
| 14 |
from transformers import CLIPTextModel, CLIPTokenizer
|
| 15 |
|
| 16 |
# custom imports
|
| 17 |
+
from model.src.datasets.dresscode import DressCodeDataset
|
| 18 |
from model.src.datasets.vitonhd import VitonHDDataset
|
| 19 |
from model.src.mgd_pipelines.mgd_pipe import MGDPipe
|
| 20 |
from model.src.mgd_pipelines.mgd_pipe_disentangled import MGDPipeDisentangled
|