lllyasviel commited on
Commit
a1689ad
·
1 Parent(s): 87246b6
Files changed (1) hide show
  1. modules/samplers_advanced.py +2 -1
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.load_model_gpu(self.refiner_model_patcher)
 
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]