From 39aeab9983d015de05ee31a5b0df5ab8bf06a297 Mon Sep 17 00:00:00 2001 From: Aedial Date: Thu, 25 May 2023 20:51:25 +0200 Subject: [PATCH] [TEST] Fix wrong image model in image gen samplers test --- tests/api/test_imagegen_samplers.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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))