phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2026-01-03 13:02:16 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

ignore ownership param

Browse source
This commit is contained in:
Ubuntu 2025-03-18 08:50:13 +00:00
parent c885015e6f
commit d7340da7a6
2 changed files with 11 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

11
llama_stack/providers/remote/post_training/nvidia/config.py
View file

@ -18,21 +18,11 @@ class NvidiaPostTrainingConfig(BaseModel):
description="The NVIDIA API key.",
)
user_id: Optional[str] = Field(
default_factory=lambda: os.getenv("NVIDIA_USER_ID", "llama-stack-user"),
description="The NVIDIA user ID.",
)
dataset_namespace: Optional[str] = Field(
default_factory=lambda: os.getenv("NVIDIA_DATASET_NAMESPACE", "default"),
description="The NVIDIA dataset namespace.",
)
access_policies: Optional[dict] = Field(
default_factory=lambda: os.getenv("NVIDIA_ACCESS_POLICIES", {"arbitrary": "json"}),
description="The NVIDIA access policies.",
)
project_id: Optional[str] = Field(
default_factory=lambda: os.getenv("NVIDIA_PROJECT_ID", "test-example-model@v1"),
description="The NVIDIA project ID.",
@ -64,7 +54,6 @@ class NvidiaPostTrainingConfig(BaseModel):
def sample_run_config(cls, **kwargs) -> Dict[str, Any]:
return {
"api_key": "${env.NVIDIA_API_KEY:}",
"user_id": "${env.NVIDIA_USER_ID:llama-stack-user}",
"dataset_namespace": "${env.NVIDIA_DATASET_NAMESPACE:default}",
"project_id": "${env.NVIDIA_PROJECT_ID:test-project}",
"customizer_url": "${env.NVIDIA_CUSTOMIZER_URL:http://nemo.test}",