remove alphas cumprod fix from samplers_common

This commit is contained in:
drhead 2024-02-26 23:46:10 -05:00 committed by GitHub
parent 94f23d00a7
commit 4dae91a1fe
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -181,10 +181,8 @@ def apply_refiner(cfg_denoiser):
cfg_denoiser.p.extra_generation_params['Refiner'] = refiner_checkpoint_info.short_title cfg_denoiser.p.extra_generation_params['Refiner'] = refiner_checkpoint_info.short_title
cfg_denoiser.p.extra_generation_params['Refiner switch at'] = refiner_switch_at cfg_denoiser.p.extra_generation_params['Refiner switch at'] = refiner_switch_at
alphas_cumprod = cfg_denoiser.p.sd_model.alphas_cumprod
with sd_models.SkipWritingToConfig(): with sd_models.SkipWritingToConfig():
sd_models.reload_model_weights(info=refiner_checkpoint_info) sd_models.reload_model_weights(info=refiner_checkpoint_info)
cfg_denoiser.p.sd_model.alphas_cumprod = alphas_cumprod
devices.torch_gc() devices.torch_gc()
cfg_denoiser.p.setup_conds() cfg_denoiser.p.setup_conds()