fix variable

This commit is contained in:
drhead 2023-12-02 14:09:18 -05:00
parent dc1adeecdd
commit 78acdcf677

View File

@ -374,7 +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 model.alphas_cumprod_original = model.alphas_cumprod
devices.dtype_unet = torch.float32 devices.dtype_unet = torch.float32
timer.record("apply float()") timer.record("apply float()")
else: else: