hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-04 11:12:35 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Use closing() with processing classes everywhere

Browse Source 
Follows up on #11569
This commit is contained in:
Aarni Koskela
2023-07-10 20:08:23 +03:00
parent bcb6ad5fab
commit 44c27ebc73
4 changed files with 24 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
modules/hypernetworks/hypernetwork.py
View File

@@ -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)