SunderAli17 commited on
Commit
ba4f1af
1 Parent(s): 810f045

Update toonmage/pipeline.py

Browse files
Files changed (1) hide show
  1. toonmage/pipeline.py +2 -2
toonmage/pipeline.py CHANGED
@@ -18,8 +18,8 @@ from safetensors.torch import load_file
18
  from torchvision.transforms import InterpolationMode
19
  from torchvision.transforms.functional import normalize, resize
20
 
21
- from eva_clip import create_model_and_transforms
22
- from eva_clip.constants import OPENAI_DATASET_MEAN, OPENAI_DATASET_STD
23
  from toonmage.encoders import IDEncoder
24
  from toonmage.utils import is_torch2_available
25
 
 
18
  from torchvision.transforms import InterpolationMode
19
  from torchvision.transforms.functional import normalize, resize
20
 
21
+ from evaclip import create_model_and_transforms
22
+ from evaclip.constants import OPENAI_DATASET_MEAN, OPENAI_DATASET_STD
23
  from toonmage.encoders import IDEncoder
24
  from toonmage.utils import is_torch2_available
25