Use closing() with processing classes everywhere

Follows up on #11569

1 files changed, 4 insertions, 2 deletions

diff --git a/modules/hypernetworks/hypernetwork.py b/modules/hypernetworks/hypernetwork.py
index 51941c11..79670b87 100644
--- a/modules/hypernetworks/hypernetwork.py
+++ b/modules/hypernetworks/hypernetwork.py
@@ -3,6 +3,7 @@ import glob
 import html

 import os

 import inspect

+from contextlib import closing

 

 import modules.textual_inversion.dataset

 import torch

@@ -711,8 +712,9 @@ def train_hypernetwork(id_task, hypernetwork_name, learn_rate, batch_size, gradi
 

                     preview_text = p.prompt

 

-                    processed = processing.process_images(p)

-                    image = processed.images[0] if len(processed.images) > 0 else None

+                    with closing(p):

+                        processed = processing.process_images(p)

+                        image = processed.images[0] if len(processed.images) > 0 else None

 

                     if unload:

                         shared.sd_model.cond_stage_model.to(devices.cpu)