CodeCow/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-06-13 21:01:06 +08:00
Code Issues Actions 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:
model.float()
model.alphas_cumprod_original = alphas_cumprod
devices.dtype_unet = torch.float32
timer.record("apply float()")
else: