fix variable
This commit is contained in:
parent
dc1adeecdd
commit
78acdcf677
|
@ -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:
|
||||||
|
|
Loading…
Reference in New Issue