batch processing run register after script

Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2026-08-23 11:34:34 +02:00
parent c5e29e3d7b
commit f4aa9d1b86
2 changed files with 2 additions and 1 deletions
+1
View File
@@ -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:
+1 -1
View File
@@ -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)