From a07d542400806efb8531828b69ee3e26b2848ab1 Mon Sep 17 00:00:00 2001 From: Ean Garvey <87458719+monorimet@users.noreply.github.com> Date: Tue, 7 Nov 2023 15:55:30 -0600 Subject: [PATCH] (Studio) Disable SD tunings and sub-model downloads (#1944) * sets --no-use_tuned and --import_mlir as defaults in SHARK Studio. --- apps/stable_diffusion/src/utils/stable_args.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/apps/stable_diffusion/src/utils/stable_args.py b/apps/stable_diffusion/src/utils/stable_args.py index 2eb68d6338..58124c14a6 100644 --- a/apps/stable_diffusion/src/utils/stable_args.py +++ b/apps/stable_diffusion/src/utils/stable_args.py @@ -308,7 +308,7 @@ def is_valid_file(arg): p.add_argument( "--import_mlir", - default=False, + default=True, action=argparse.BooleanOptionalAction, help="Imports the model from torch module to shark_module otherwise " "downloads the model from shark_tank.", @@ -331,7 +331,7 @@ def is_valid_file(arg): p.add_argument( "--use_tuned", - default=True, + default=False, action=argparse.BooleanOptionalAction, help="Download and use the tuned version of the model if available.", )