diff --git a/modules/img2img.py b/modules/img2img.py index c6210cd16..70d7e48cc 100644 --- a/modules/img2img.py +++ b/modules/img2img.py @@ -338,6 +338,7 @@ def img2img(id_task: str, state: str, mode: int, if p.is_batch: process_batch(p, img2img_batch_files, img2img_batch_input_dir, img2img_batch_output_dir, img2img_batch_inpaint_mask_dir, args) processed = processing.get_processed(p, [], p.seed, "") + processed = scripts_manager.scripts_img2img.after(p, processed, *args) else: processed = scripts_manager.scripts_img2img.run(p, *args) if processed is None: diff --git a/pipelines/sefi/convert_sefi_to_diffusers.py b/pipelines/sefi/convert_sefi_to_diffusers.py index fca38a5a5..08d88cc61 100644 --- a/pipelines/sefi/convert_sefi_to_diffusers.py +++ b/pipelines/sefi/convert_sefi_to_diffusers.py @@ -317,7 +317,7 @@ def convert_scheduler(root: Path, output: Path): torch.testing.assert_close(scheduler.sigmas, source_scheduler.sigmas, rtol=0.0, atol=0.0) torch.testing.assert_close(scheduler.timesteps, source_scheduler.timesteps, rtol=0.0, atol=0.0) - num_train_timesteps = int(scheduler.config.num_train_timesteps) + num_train_timesteps = int(scheduler.config.num_train_timesteps) # pylint: disable=no-member expected_sigmas = torch.linspace(1.0, 1.0 / num_train_timesteps, num_train_timesteps) torch.testing.assert_close(scheduler.sigmas, expected_sigmas, rtol=0.0, atol=1e-7) torch.testing.assert_close(scheduler.timesteps, expected_sigmas * num_train_timesteps, rtol=0.0, atol=1e-4)