Commit 57e59c14 authored by AUTOMATIC1111's avatar AUTOMATIC1111
Browse files

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

This reverts commit 0815c45b.
parent 0815c45b

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.device):

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

        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