Commit
·
a1689ad
1
Parent(s):
87246b6
ram fix
Browse files
modules/samplers_advanced.py
CHANGED
@@ -152,7 +152,8 @@ class KSamplerWithRefiner:
|
|
152 |
noise.shape[3], noise.shape[2], self.device, "negative")
|
153 |
|
154 |
def refiner_switch():
|
155 |
-
comfy.model_management.
|
|
|
156 |
self.model_denoise.inner_model = self.refiner_model_denoise.inner_model
|
157 |
for i in range(len(positive)):
|
158 |
positive[i] = refiner_positive[i]
|
|
|
152 |
noise.shape[3], noise.shape[2], self.device, "negative")
|
153 |
|
154 |
def refiner_switch():
|
155 |
+
comfy.model_management.load_models_gpu([self.refiner_model_patcher], comfy.model_management.batch_area_memory(
|
156 |
+
noise.shape[0] * noise.shape[2] * noise.shape[3]))
|
157 |
self.model_denoise.inner_model = self.refiner_model_denoise.inner_model
|
158 |
for i in range(len(positive)):
|
159 |
positive[i] = refiner_positive[i]
|