Update StableDiffusionWebUI.py

shellward · Sep 11, 2022 · cc63233 · cc63233
1 parent 3d58952
commit cc63233
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/StableDiffusionWebUI.py b/StableDiffusionWebUI.py
@@ -215,7 +215,7 @@ def validate(prompt, steps, cfg_scale, width, height, num_batches, batch_size, i
         checkHeight(height)
         checkNumBatches(num_batches)
         checkBatchSize(batch_size)
-        if function_index == FunctionIndex.IMAGE_TO_IMAGE or FunctionIndex.IMAGE_TO_IMAGE_WITH_MASK:
+        if function_index == FunctionIndex.IMAGE_TO_IMAGE or function_index == FunctionIndex.IMAGE_TO_IMAGE_WITH_MASK:
             checkImageString(image_string)
             checkStrength(strength)
             if function_index== FunctionIndex.IMAGE_TO_IMAGE_WITH_MASK: