This commit is contained in:
lllyasviel 2023-10-12 18:55:33 -07:00 committed by GitHub
parent 3b9266e848
commit c4118d2418
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions

View File

@ -134,6 +134,9 @@ def sample_hacked(model, noise, positive, negative, cfg, device, sampler, sigmas
if step == refiner_switch_step and current_refiner is not None:
refiner_switch()
if callback is not None:
# residual_noise_preview = x - x0
# residual_noise_preview /= residual_noise_preview.std()
# residual_noise_preview *= x0.std()
callback(step, x0, x, total_steps)
samples = sampler.sample(model_wrap, sigmas, extra_args, callback_wrap, noise, latent_image, denoise_mask, disable_pbar)