Skip to content

Commit

Permalink
(Studio) Disable SD tunings and sub-model downloads (#1944)
Browse files Browse the repository at this point in the history
* sets --no-use_tuned and --import_mlir as defaults in SHARK Studio.
  • Loading branch information
monorimet authored Nov 7, 2023
1 parent ad55cb6 commit a07d542
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions apps/stable_diffusion/src/utils/stable_args.py
Original file line number Diff line number Diff line change
Expand Up @@ -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.",
Expand All @@ -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.",
)
Expand Down

0 comments on commit a07d542

Please sign in to comment.