remove batch_size from img2img (#468)

This commit is contained in:
Thomas Mello 2022-09-02 00:03:12 +03:00 committed by GitHub
parent 561fb3b8ce
commit 6a054b3615
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1135,12 +1135,14 @@ class Flagging(gr.FlaggingCallback):
def img2img(prompt: str, image_editor_mode: str, init_info, mask_mode: str, mask_blur_strength: int, ddim_steps: int, sampler_name: str,
toggles: List[int], realesrgan_model_name: str, n_iter: int, batch_size: int, cfg_scale: float, denoising_strength: float,
toggles: List[int], realesrgan_model_name: str, n_iter: int, cfg_scale: float, denoising_strength: float,
seed: int, height: int, width: int, resize_mode: int, fp):
outpath = opt.outdir_img2img or opt.outdir or "outputs/img2img-samples"
err = False
seed = seed_to_int(seed)
batch_size = 1
prompt_matrix = 0 in toggles
normalize_prompt_weights = 1 in toggles
loopback = 2 in toggles