feat: use args temp path if given
This commit is contained in:
parent
161b988522
commit
1779fa0f3a
|
|
@ -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
|
||||
|
|
|
|||
|
|
@ -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)
|
||||
|
|
|
|||
Loading…
Reference in New Issue