aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorbrkirch <brkirch@users.noreply.github.com>2023-08-21 21:36:17 +0000
committerbrkirch <brkirch@users.noreply.github.com>2023-08-21 21:36:17 +0000
commit299b8096bc19c8a619fbeef735cb90544350243c (patch)
treecbc35adf9f5b6859271934aa8e15ae5889b3781e
parent953c3eab7b3b952f7e96d728413a531d7fb521a2 (diff)
downloadstable-diffusion-webui-gfx803-299b8096bc19c8a619fbeef735cb90544350243c.tar.gz
stable-diffusion-webui-gfx803-299b8096bc19c8a619fbeef735cb90544350243c.tar.bz2
stable-diffusion-webui-gfx803-299b8096bc19c8a619fbeef735cb90544350243c.zip
Make Gradio temp directory if it doesn't exist
Gradio normally creates the temp directory in `pil_to_temp_file()` (https://github.com/gradio-app/gradio/blob/861d752a83da0f95e9f79173069b69eababeed39/gradio/components/base.py#L313) but since the Gradio implementation of `pil_to_temp_file()` is replaced with `save_pil_to_file()`, the Gradio temp directory should also be created by `save_pil_to_file()` when necessary.
-rw-r--r--modules/ui_tempdir.py2
1 files changed, 2 insertions, 0 deletions
diff --git a/modules/ui_tempdir.py b/modules/ui_tempdir.py
index 506017e5..85015db5 100644
--- a/modules/ui_tempdir.py
+++ b/modules/ui_tempdir.py
@@ -44,6 +44,8 @@ def save_pil_to_file(self, pil_image, dir=None, format="png"):
if shared.opts.temp_dir != "":
dir = shared.opts.temp_dir
+ else:
+ os.makedirs(dir, exist_ok=True)
use_metadata = False
metadata = PngImagePlugin.PngInfo()