From 63f5723449a1f1591e9fd5ba685be2b142aeef69 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?C=C3=A9dric=20Deltheil?= Date: Sun, 24 Sep 2023 22:26:31 +0200 Subject: [PATCH] test_concepts: silent static type checker error --- tests/foundationals/clip/test_concepts.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/foundationals/clip/test_concepts.py b/tests/foundationals/clip/test_concepts.py index bf42fe8..d297f01 100644 --- a/tests/foundationals/clip/test_concepts.py +++ b/tests/foundationals/clip/test_concepts.py @@ -44,8 +44,9 @@ def our_encoder_with_new_concepts( @pytest.fixture(scope="module") def ref_sd15_with_new_concepts( runwayml_weights_path: Path, test_textual_inversion_path: Path, test_device: torch.device -): +) -> StableDiffusionPipeline: pipe = StableDiffusionPipeline.from_pretrained(runwayml_weights_path).to(test_device) # type: ignore + assert isinstance(pipe, StableDiffusionPipeline) pipe.load_textual_inversion(test_textual_inversion_path / "cat-toy") # type: ignore pipe.load_textual_inversion(test_textual_inversion_path / "gta5-artwork") # type: ignore return pipe