Merge pull request #11720 from akx/closing (910d4f61) · Commits · github_fork / Stable Diffusion Webui

modules/hypernetworks/hypernetwork.py

+4 −2

Original line number	Diff line number	Diff line
		@@ -3,6 +3,7 @@ import glob
		import html
		import os
		import inspect
		from contextlib import closing

		import modules.textual_inversion.dataset
		import torch
		@@ -711,6 +712,7 @@ def train_hypernetwork(id_task, hypernetwork_name, learn_rate, batch_size, gradi

		preview_text = p.prompt

		with closing(p):
		processed = processing.process_images(p)
		image = processed.images[0] if len(processed.images) > 0 else None

+10 −10

Original line number	Diff line number	Diff line
		import os
		from contextlib import closing
		from pathlib import Path

		import numpy as np
		@@ -217,6 +218,7 @@ def img2img(id_task: str, mode: int, prompt: str, negative_prompt: str, prompt_s
		if mask:
		p.extra_generation_params["Mask blur"] = mask_blur

		with closing(p):
		if is_batch:
		assert not shared.cmd_opts.hide_ui_dir_config, "Launched with --hide-ui-dir-config, batch img2img disabled"

		@@ -228,8 +230,6 @@ def img2img(id_task: str, mode: int, prompt: str, negative_prompt: str, prompt_s
		if processed is None:
		processed = process_images(p)

		p.close()

		shared.total_tqdm.clear()

		generation_info_js = processed.js()

+4 −2

Original line number	Diff line number	Diff line
		import os
		from collections import namedtuple
		from contextlib import closing

		import torch
		import tqdm
		@@ -584,6 +585,7 @@ def train_embedding(id_task, embedding_name, learn_rate, batch_size, gradient_st

		preview_text = p.prompt

		with closing(p):
		processed = processing.process_images(p)
		image = processed.images[0] if len(processed.images) > 0 else None

+6 −5

Original line number	Diff line number	Diff line
		from contextlib import closing

		import modules.scripts
		from modules import sd_samplers, processing
		from modules.generation_parameters_copypaste import create_override_settings_dict
		@@ -53,13 +55,12 @@ def txt2img(id_task: str, prompt: str, negative_prompt: str, prompt_styles, step
		if cmd_opts.enable_console_prompts:
		print(f"\ntxt2img: {prompt}", file=shared.progress_print_out)

		with closing(p):
		processed = modules.scripts.scripts_txt2img.run(p, *args)

		if processed is None:
		processed = processing.process_images(p)

		p.close()

		shared.total_tqdm.clear()

		generation_info_js = processed.js()