OPENGIT/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-02-01 03:03:00 +08:00
Code Issues Packages Projects Releases 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
Show Stats 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: if shared.cmd_opts.no_half:
model.float() model.float()
model.alphas_cumprod_original = alphas_cumprod
devices.dtype_unet = torch.float32 devices.dtype_unet = torch.float32
timer.record("apply float()") timer.record("apply float()")
else: else: