Add interrupt button to preprocessing (c8c40c8a) · Commits · github_fork / Stable Diffusion Webui

modules/textual_inversion/ui.py

+1 −1

Original line number	Original line	Diff line number	Diff line
	@@ -18,7 +18,7 @@ def create_embedding(name, initialization_text, nvpt, overwrite_old):
	def preprocess(*args):		def preprocess(*args):
	modules.textual_inversion.preprocess.preprocess(*args)		modules.textual_inversion.preprocess.preprocess(*args)

	return "Preprocessing finished.", ""		return f"Preprocessing {'interrupted' if shared.state.interrupted else 'finished'}.", ""


	def train_embedding(*args):		def train_embedding(*args):

+9 −1

Original line number	Original line	Diff line number	Diff line
	@@ -1249,6 +1249,8 @@ def create_ui(wrap_gradio_gpu_call):
	gr.HTML(value="")		gr.HTML(value="")

	with gr.Column():		with gr.Column():
			with gr.Row():
			interrupt_preprocessing = gr.Button("Interrupt")
	run_preprocess = gr.Button(value="Preprocess", variant='primary')		run_preprocess = gr.Button(value="Preprocess", variant='primary')

	process_split.change(		process_split.change(
	@@ -1422,6 +1424,12 @@ def create_ui(wrap_gradio_gpu_call):
	outputs=[],		outputs=[],
	)		)

			interrupt_preprocessing.click(
			fn=lambda: shared.state.interrupt(),
			inputs=[],
			outputs=[],
			)

	def create_setting_component(key, is_quicksettings=False):		def create_setting_component(key, is_quicksettings=False):
	def fun():		def fun():
	return opts.data[key] if key in opts.data else opts.data_labels[key].default		return opts.data[key] if key in opts.data else opts.data_labels[key].default