|
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.") |
|
|
|
|
|
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.") |
|
|
|
args_parser.parser.add_argument("--disable-metadata", action='store_true', |
|
help="Disables saving metadata to images.") |
|
|
|
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() |
|
|
|
|
|
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 |
|
|
