mirror of
https://github.com/lllyasviel/Fooocus.git
synced 2026-08-16 13:13:16 +02:00
feat: use args temp path if given
This commit is contained in:
@@ -1,5 +1,7 @@
|
|||||||
import ldm_patched.modules.args_parser as args_parser
|
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("--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.")
|
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:
|
if args_parser.args.disable_analytics:
|
||||||
import os
|
import os
|
||||||
os.environ["GRADIO_ANALYTICS_ENABLED"] = "False"
|
os.environ["GRADIO_ANALYTICS_ENABLED"] = "False"
|
||||||
|
|
||||||
if args_parser.args.disable_in_browser:
|
if args_parser.args.disable_in_browser:
|
||||||
args_parser.args.in_browser = False
|
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
|
args = args_parser.args
|
||||||
|
|||||||
@@ -3,11 +3,10 @@ import args_manager
|
|||||||
import modules.config
|
import modules.config
|
||||||
import json
|
import json
|
||||||
import urllib.parse
|
import urllib.parse
|
||||||
import tempfile
|
|
||||||
|
|
||||||
from PIL import Image
|
from PIL import Image
|
||||||
from modules.util import generate_temp_filename
|
from modules.util import generate_temp_filename
|
||||||
|
from tempfile import gettempdir
|
||||||
|
|
||||||
log_cache = {}
|
log_cache = {}
|
||||||
|
|
||||||
@@ -20,7 +19,7 @@ def get_current_html_path():
|
|||||||
|
|
||||||
|
|
||||||
def log(img, dic) -> str:
|
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')
|
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)
|
os.makedirs(os.path.dirname(local_temp_filename), exist_ok=True)
|
||||||
Image.fromarray(img).save(local_temp_filename)
|
Image.fromarray(img).save(local_temp_filename)
|
||||||
|
|||||||
Reference in New Issue
Block a user