1580158
/
stable-diffusion-webui-stable-horde
mirror of https://github.com/natanjunges/stable-diffusion-webui-stable-horde.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add support to pix2pix

main
Natan Junges 2023-02-01 14:37:01 -03:00 committed by GitHub
parent 393c6ea784
commit 2af4a48445
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/main.py
View File

@ -367,7 +367,7 @@ class Main(SettingsManager, scripts.Script):
def process_batch_horde(self, p, model, nsfw, shared_laion, seed_variation, post_processing, prompt, negative_prompt, seed):
try:
assert model != "stable_diffusion_inpainting" or self.is_img2img and p.image_mask is not None, "Model stable_diffusion_inpainting can only be used for inpainting"
assert model != "Stable Diffusion 2 Depth" or self.is_img2img and p.image_mask is None, "Model Stable Diffusion 2 Depth can only be used for img2img"
assert model not in {"Stable Diffusion 2 Depth", "pix2pix"} or self.is_img2img and p.image_mask is None, "Model {} can only be used for img2img".format(model)
except AssertionError as e:
raise StableHordeGenerateError(str(e))