From 3f82820612ade9bbfd6d9c31c6482699e7587385 Mon Sep 17 00:00:00 2001
From: AUTOMATIC1111 <16777216c@gmail.com>
Date: Mon, 7 Aug 2023 08:16:20 +0300
Subject: [PATCH] apply unet overrides after switching model

---
 modules/sd_models.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/modules/sd_models.py b/modules/sd_models.py
index 981aa93d7..fce278612 100644
--- a/modules/sd_models.py
+++ b/modules/sd_models.py
@@ -715,6 +715,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