SD-online/Fooocus-Docker

1import ldm_patched.modules.args_parser as args_parser2import os3 4from tempfile import gettempdir5 6args_parser.parser.add_argument("--share", action='store_true', help="Set whether to share on Gradio.")7 8args_parser.parser.add_argument("--preset", type=str, default=None, help="Apply specified UI preset.")9args_parser.parser.add_argument("--disable-preset-selection", action='store_true',10                                help="Disables preset selection in Gradio.")11 12args_parser.parser.add_argument("--language", type=str, default='default',13                                help="Translate UI using json files in [language] folder. "14                                  "For example, [--language example] will use [language/example.json] for translation.")15 16# For example, https://github.com/lllyasviel/Fooocus/issues/84917args_parser.parser.add_argument("--disable-offload-from-vram", action="store_true",18                                help="Force loading models to vram when the unload can be avoided. "19                                  "Some Mac users may need this.")20 21args_parser.parser.add_argument("--theme", type=str, help="launches the UI with light or dark theme", default=None)22args_parser.parser.add_argument("--disable-image-log", action='store_true',23                                help="Prevent writing images and logs to hard drive.")24 25args_parser.parser.add_argument("--disable-analytics", action='store_true',26                                help="Disables analytics for Gradio.")27 28args_parser.parser.add_argument("--disable-metadata", action='store_true',29                                help="Disables saving metadata to images.")30 31args_parser.parser.add_argument("--disable-preset-download", action='store_true',32                                help="Disables downloading models for presets", default=False)33 34args_parser.parser.add_argument("--always-download-new-model", action='store_true',35                                help="Always download newer models ", default=False)36 37args_parser.parser.set_defaults(38    disable_cuda_malloc=True,39    in_browser=True,40    port=None41)42 43args_parser.args = args_parser.parser.parse_args()44 45# (Disable by default because of issues like https://github.com/lllyasviel/Fooocus/issues/724)46args_parser.args.always_offload_from_vram = not args_parser.args.disable_offload_from_vram47 48if args_parser.args.disable_analytics:49    import os50    os.environ["GRADIO_ANALYTICS_ENABLED"] = "False"51 52if args_parser.args.disable_in_browser:53    args_parser.args.in_browser = False54 55args = args_parser.args56