mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-07-27 06:28:50 +00:00
feat: disable post training in starter but keep the CI test
The default starter distro won't enable post_training by default, to activate the provider run the distro this env variable ENABLE_HUGGINGFACE=huggingface Signed-off-by: Sébastien Han <seb@redhat.com>
This commit is contained in:
parent
49fa26949a
commit
3ddf8c1b84
3 changed files with 7 additions and 2 deletions
|
@ -179,7 +179,7 @@ providers:
|
|||
sqlite_db_path: ${env.SQLITE_STORE_DIR:=~/.llama/distributions/starter}/trace_store.db
|
||||
otel_exporter_otlp_endpoint: ${env.OTEL_EXPORTER_OTLP_ENDPOINT:=}
|
||||
post_training:
|
||||
- provider_id: huggingface
|
||||
- provider_id: ${env.ENABLE_HUGGINGFACE:=__disabled__}
|
||||
provider_type: inline::huggingface
|
||||
config:
|
||||
checkpoint_format: huggingface
|
||||
|
|
|
@ -281,7 +281,7 @@ def get_distribution_template() -> DistributionTemplate:
|
|||
config=SentenceTransformersInferenceConfig.sample_run_config(),
|
||||
)
|
||||
post_training_provider = Provider(
|
||||
provider_id="huggingface",
|
||||
provider_id="${env.ENABLE_HUGGINGFACE:=__disabled__}",
|
||||
provider_type="inline::huggingface",
|
||||
config=HuggingFacePostTrainingConfig.sample_run_config(f"~/.llama/distributions/{name}"),
|
||||
)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue