Skip to content

S
Stable Diffusion Webui
Commit 57e59c14 authored by AUTOMATIC1111's avatar AUTOMATIC1111
Browse files
Options

Revert "send weights to target device instead of CPU memory" 

This reverts commit 0815c45b.
parent 0815c45b
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
modules/sd_models.py
View file @ 57e59c14
...@@ -579,7 +579,7 @@ def load_model(checkpoint_info=None, already_loaded_state_dict=None): ...@@ -579,7 +579,7 @@ def load_model(checkpoint_info=None, already_loaded_state_dict=None):
timer.record("create model") timer.record("create model")
with sd_disable_initialization.LoadStateDictOnMeta(state_dict, devices.device): with sd_disable_initialization.LoadStateDictOnMeta(state_dict, devices.cpu):
load_model_weights(sd_model, checkpoint_info, state_dict, timer) load_model_weights(sd_model, checkpoint_info, state_dict, timer)
timer.record("load weights from state dict") timer.record("load weights from state dict")
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment