Spaces:
Running
on
L4
Running
on
L4
Update MagicQuill/brushnet_nodes.py
Browse files
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()
|