feat: use args temp path if given

This commit is contained in:
Manuel Schmid 2024-01-14 21:19:01 +01:00
parent 161b988522
commit 1779fa0f3a
No known key found for this signature in database
GPG Key ID: 32C4F7569B40B84B
2 changed files with 8 additions and 3 deletions

View File

@ -1,5 +1,7 @@
import ldm_patched.modules.args_parser as args_parser
import os
from tempfile import gettempdir
args_parser.parser.add_argument("--share", action='store_true', help="Set whether to share on Gradio.")
args_parser.parser.add_argument("--preset", type=str, default=None, help="Apply specified UI preset.")
@ -34,7 +36,11 @@ args_parser.args.always_offload_from_vram = not args_parser.args.disable_offload
if args_parser.args.disable_analytics:
import os
os.environ["GRADIO_ANALYTICS_ENABLED"] = "False"
if args_parser.args.disable_in_browser:
args_parser.args.in_browser = False
if args_parser.args.temp_path is None:
args_parser.args.temp_path = os.path.join(gettempdir(), 'Fooocus')
args = args_parser.args

View File

@ -3,11 +3,10 @@ import args_manager
import modules.config
import json
import urllib.parse
import tempfile
from PIL import Image
from modules.util import generate_temp_filename
from tempfile import gettempdir
log_cache = {}
@ -20,7 +19,7 @@ def get_current_html_path():
def log(img, dic) -> str:
path_outputs = tempfile.gettempdir() if args_manager.args.disable_image_log else modules.config.path_outputs
path_outputs = args_manager.args.temp_path if args_manager.args.disable_image_log else modules.config.path_outputs
date_string, local_temp_filename, only_name = generate_temp_filename(folder=path_outputs, extension='png')
os.makedirs(os.path.dirname(local_temp_filename), exist_ok=True)
Image.fromarray(img).save(local_temp_filename)