diff --git a/tests/api/test_imagegen_samplers.py b/tests/api/test_imagegen_samplers.py index fb249ee..704675b 100644 --- a/tests/api/test_imagegen_samplers.py +++ b/tests/api/test_imagegen_samplers.py @@ -14,7 +14,9 @@ sampler_xfail = pytest.mark.xfail(True, raises=NovelAIError, reason="The sampler doesn't currently work") models = list(ImageModel) -models.remove(ImageModel.Anime_Inpainting) +models.remove(ImageModel.Inpainting_Anime_Full) +models.remove(ImageModel.Inainting_Anime_Curated) +models.remove(ImageModel.Inpainting_Furry) samplers = list(ImageSampler) model_samplers = list(itertools.product(models, samplers))