LiuZichen commited on
Commit
fd548a8
·
verified ·
1 Parent(s): 4420ac6

Update MagicQuill/brushnet_nodes.py

Browse files
Files changed (1) hide show
  1. MagicQuill/brushnet_nodes.py +2 -2
MagicQuill/brushnet_nodes.py CHANGED
@@ -287,7 +287,7 @@ class PowerPaint:
287
  # unload vae and CLIPs
288
  del vae
289
  del clip
290
- for loaded_model in comfy.model_management.current_loaded_models:
291
  if type(loaded_model.model.model) in ModelsToUnload:
292
  comfy.model_management.current_loaded_models.remove(loaded_model)
293
  loaded_model.model_unload()
@@ -375,7 +375,7 @@ class BrushNet:
375
 
376
  # unload vae
377
  del vae
378
- for loaded_model in comfy.model_management.current_loaded_models:
379
  if type(loaded_model.model.model) in ModelsToUnload:
380
  comfy.model_management.current_loaded_models.remove(loaded_model)
381
  loaded_model.model_unload()
 
287
  # unload vae and CLIPs
288
  del vae
289
  del clip
290
+ for loaded_model in list(comfy.model_management.current_loaded_models):
291
  if type(loaded_model.model.model) in ModelsToUnload:
292
  comfy.model_management.current_loaded_models.remove(loaded_model)
293
  loaded_model.model_unload()
 
375
 
376
  # unload vae
377
  del vae
378
+ for loaded_model in list(comfy.model_management.current_loaded_models):
379
  if type(loaded_model.model.model) in ModelsToUnload:
380
  comfy.model_management.current_loaded_models.remove(loaded_model)
381
  loaded_model.model_unload()