mirror of
https://github.com/vladmandic/automatic
synced 2026-09-19 09:14:35 +02:00
batch processing run register after script
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
@@ -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:
|
||||
|
||||
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user