Fooocus/args_manager.py
lllyasviel 0a2c1c4cbf
amd support (#607)
* amd support

* amd support
2023-10-09 15:47:46 -07:00

13 lines
359 B
Python

from comfy.options import enable_args_parsing
enable_args_parsing(False)
import comfy.cli_args as comfy_cli
comfy_cli.parser.add_argument("--share", action='store_true', help="Set whether to share on Gradio.")
comfy_cli.args = comfy_cli.parser.parse_args()
comfy_cli.args.disable_cuda_malloc = True
comfy_cli.args.auto_launch = True
args = comfy_cli.args