Commit 6e7828e1 authored by AUTOMATIC1111's avatar AUTOMATIC1111
Browse files

apply unet overrides after switching model

parent c96e4750

modules/sd_models.py

+1 −0
Original line number Diff line number Diff line
@@ -699,6 +699,7 @@ def reload_model_weights(sd_model=None, info=None): 
    print(f"Weights loaded in {timer.summary()}.")



    model_data.set_sd_model(sd_model)

    sd_unet.apply_unet()



    return sd_model

CRA Git | Maintained and supported by SUSTech CRA and CCSE