phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-29 15:23:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

more robust image type

Browse source
This commit is contained in:
Xi Yan 2024-09-28 17:44:45 -07:00
parent b646167d94
commit 12c0860cae
1 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

7
llama_stack/cli/stack/build.py
View file

@ -74,8 +74,8 @@ class StackBuild(Subcommand):
self.parser.add_argument( self.parser.add_argument(
"--image-type", "--image-type",
type=str, type=str,
help="Image Type to use for the build. This can be either conda or docker. If not specified, will use conda by default", help="Image Type to use for the build. This can be either conda or docker. If not specified, will use the image type from the template config.",
default="conda", choices=["conda", "docker"],
) )
def _run_stack_build_command_from_build_config( def _run_stack_build_command_from_build_config(
@ -183,7 +183,8 @@ class StackBuild(Subcommand):
with open(build_path, "r") as f: with open(build_path, "r") as f:
build_config = BuildConfig(**yaml.safe_load(f)) build_config = BuildConfig(**yaml.safe_load(f))
build_config.name = args.name build_config.name = args.name
build_config.image_type = args.image_type if args.image_type:
build_config.image_type = args.image_type
self._run_stack_build_command_from_build_config(build_config) self._run_stack_build_command_from_build_config(build_config)
return return