phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-13 04:22:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

added platform for docker image

Browse source
This commit is contained in:
picografix 2024-10-17 12:32:07 +05:30
parent c4d5d6bb91
commit 0c75f2cf31
4 changed files with 24 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

6
llama_stack/distribution/build.py
View file

@ -35,7 +35,6 @@ class ImageType(Enum):
docker = "docker"
conda = "conda"
class Dependencies(BaseModel):
pip_packages: List[str]
docker_image: Optional[str] = None
@ -91,10 +90,12 @@ def build_image(build_config: BuildConfig, build_file_path: Path):
script = pkg_resources.resource_filename(
"llama_stack", "distribution/build_container.sh"
)
args = [
script,
build_config.name,
package_deps.docker_image,
build_config.platform,
str(build_file_path),
str(BUILDS_BASE_DIR / ImageType.docker.value),
" ".join(deps),
@ -107,8 +108,9 @@ def build_image(build_config: BuildConfig, build_file_path: Path):
script,
build_config.name,
str(build_file_path),
" ".join(deps),
" ".join(deps)
]
if special_deps:
args.append("#".join(special_deps))