diff --git a/modules/sd_models.py b/modules/sd_models.py
index d65735e3a..53c1df54c 100644
--- a/modules/sd_models.py
+++ b/modules/sd_models.py
@@ -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