diff --git a/tests/test_consistency.py b/tests/test_consistency.py index a90672556..da966f1b6 100644 --- a/tests/test_consistency.py +++ b/tests/test_consistency.py @@ -1,9 +1,10 @@ -import clip import numpy as np import pytest import torch from PIL import Image +import clip + @pytest.mark.parametrize("model_name", clip.available_models()) def test_consistency(model_name):