fix: missed elif - bad unittest coverage

provos · Jun 14, 2024 · b9080ed · b9080ed
1 parent 8276700
commit b9080ed
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/inpainting.py b/inpainting.py
@@ -197,7 +197,7 @@ def inpaint(self, prompt, negative_prompt, init_image, mask_image,
             image = self.inpaint_image2image(
                 resize_init_image, prompt, negative_prompt,
                 strength, guidance_scale, num_inference_steps, seed)
-        if self.model in self.MODELS:
+        elif self.model in self.MODELS:
             image = self.inpaint_diffusers(
                 resize_init_image, resize_mask_image,
                 prompt, negative_prompt,