phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-17 18:52:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add ollama/pull-models.sh

Browse source
This commit is contained in:
Ashwin Bharambe 2024-11-18 10:57:20 -08:00
parent fa1d29cfdc
commit 1ecaf2cb3c
16 changed files with 305 additions and 289 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/templates/ollama/ollama.py
View file

@ -28,11 +28,11 @@ def get_distribution_template() -> DistributionTemplate:
inference_model = ModelInput(
model_id="${env.INFERENCE_MODEL}",
provider_id="ollama-inference",
provider_id="ollama",
)
safety_model = ModelInput(
model_id="${env.SAFETY_MODEL}",
provider_id="ollama-safety",
provider_id="ollama",
)
return DistributionTemplate(