Update webui.py

This commit is contained in:
jjisnow 2022-09-17 16:09:53 +10:00 committed by GitHub
parent 588d6de4a8
commit b172cd56e8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions

View File

@ -9,7 +9,7 @@ from omegaconf import OmegaConf
import signal import signal
from ldm.util import instantiate_from_config from ldm.util import instantiate_from_config
run_
from modules.shared import opts, cmd_opts, state from modules.shared import opts, cmd_opts, state
import modules.shared as shared import modules.shared as shared
import modules.ui import modules.ui
@ -121,7 +121,7 @@ def webui():
txt2img=wrap_gradio_gpu_call(modules.txt2img.txt2img), txt2img=wrap_gradio_gpu_call(modules.txt2img.txt2img),
img2img=wrap_gradio_gpu_call(modules.img2img.img2img), img2img=wrap_gradio_gpu_call(modules.img2img.img2img),
run_extras=wrap_gradio_gpu_call(modules.extras.run_extras), run_extras=wrap_gradio_gpu_call(modules.extras.run_extras),
run_image_info=modules.extras.run_image_info run_pnginfo=modules.extras.run_pnginfo
) )
demo.launch( demo.launch(