Commit 4d9b0966 authored by AUTOMATIC1111's avatar AUTOMATIC1111
Browse files

additional memory improvements when switching between models of different types

parent 29d7e31d

modules/sd_models.py

+4 −1
Original line number Diff line number Diff line
@@ -582,7 +582,10 @@ def reload_model_weights(sd_model=None, info=None): 
    timer.record("find config")



    if sd_model is None or checkpoint_config != sd_model.used_config:

        del sd_model

        if sd_model is not None:

            sd_model.to(device="meta")



        devices.torch_gc()

        load_model(checkpoint_info, already_loaded_state_dict=state_dict)

        return model_data.sd_model

CRA Git | Maintained and supported by SUSTech CRA and CCSE