mirrors/stable-diffusion-webui
1
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-12-26 11:52:42 -08:00
Code Issues Projects Releases Packages Wiki Activity

Create alphas_cumprod_original on full precision path

Browse source
This commit is contained in:
drhead 2023-12-02 14:06:56 -05:00
parent 4a43334376
commit dc1adeecdd
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
modules/sd_models.py
View file

@ -374,6 +374,7 @@ def load_model_weights(model, checkpoint_info: CheckpointInfo, state_dict, timer
if shared.cmd_opts.no_half:
model.float()
model.alphas_cumprod_original = alphas_cumprod
devices.dtype_unet = torch.float32
timer.record("apply float()")
else: