d3113f5c3f
* feat: use consistent file name in gradio returns and uses filepaths instead of numpy image by saving to temp dir uses double the temp dir file storage on disk as it saves to temp dir and gradio temp dir when displaying the image, but reuses logged output image * feat: delete temp images after yielding to gradio * feat: use args temp path if given * chore: code cleanup, remove redundant if statement
53 lines
2.3 KiB
Python
53 lines
2.3 KiB
Python
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.")
|
|
|
|
args_parser.parser.add_argument("--language", type=str, default='default',
|
|
help="Translate UI using json files in [language] folder. "
|
|
"For example, [--language example] will use [language/example.json] for translation.")
|
|
|
|
# For example, https://github.com/lllyasviel/Fooocus/issues/849
|
|
args_parser.parser.add_argument("--disable-offload-from-vram", action="store_true",
|
|
help="Force loading models to vram when the unload can be avoided. "
|
|
"Some Mac users may need this.")
|
|
|
|
args_parser.parser.add_argument("--theme", type=str, help="launches the UI with light or dark theme", default=None)
|
|
args_parser.parser.add_argument("--disable-image-log", action='store_true',
|
|
help="Prevent writing images and logs to hard drive.")
|
|
|
|
args_parser.parser.add_argument("--disable-analytics", action='store_true',
|
|
help="Disables analytics for Gradio", default=False)
|
|
|
|
args_parser.parser.add_argument("--disable-preset-download", action='store_true',
|
|
help="Disables downloading models for presets", default=False)
|
|
|
|
args_parser.parser.add_argument("--always-download-new-model", action='store_true',
|
|
help="Always download newer models ", default=False)
|
|
|
|
args_parser.parser.set_defaults(
|
|
disable_cuda_malloc=True,
|
|
in_browser=True,
|
|
port=None
|
|
)
|
|
|
|
args_parser.args = args_parser.parser.parse_args()
|
|
|
|
# (Disable by default because of issues like https://github.com/lllyasviel/Fooocus/issues/724)
|
|
args_parser.args.always_offload_from_vram = not args_parser.args.disable_offload_from_vram
|
|
|
|
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
|