Commit 0815c45b authored by AUTOMATIC1111's avatar AUTOMATIC1111
Browse files

send weights to target device instead of CPU memory

parent 86221269

modules/sd_models.py

+1 −1
Original line number Diff line number Diff line
@@ -579,7 +579,7 @@ def load_model(checkpoint_info=None, already_loaded_state_dict=None): 


    timer.record("create model")



    with sd_disable_initialization.LoadStateDictOnMeta(state_dict, devices.cpu):

    with sd_disable_initialization.LoadStateDictOnMeta(state_dict, devices.device):

        load_model_weights(sd_model, checkpoint_info, state_dict, timer)

    timer.record("load weights from state dict")

CRA Git | Maintained and supported by SUSTech CRA and CCSE