phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-30 06:34:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge branch 'main' into register_custom_model

Browse source
This commit is contained in:
Rashmi Pawar 2025-04-16 13:47:33 +05:30
commit 8000b0287f
242 changed files with 221047 additions and 8397 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/providers/remote/post_training/nvidia/models.py
View file

@ -6,7 +6,7 @@
from typing import List
from llama_stack.models.llama.datatypes import CoreModelId
from llama_stack.models.llama.sku_types import CoreModelId
from llama_stack.providers.utils.inference.model_registry import (
ProviderModelEntry,
build_hf_repo_model_entry,

4
llama_stack/providers/remote/post_training/nvidia/post_training.py
View file

@ -206,10 +206,6 @@ class NvidiaPostTrainingAdapter(ModelRegistryHelper):
model: str,
checkpoint_dir: Optional[str],
algorithm_config: Optional[AlgorithmConfig] = None,
extra_json: Optional[Dict[str, Any]] = None,
params: Optional[Dict[str, Any]] = None,
headers: Optional[Dict[str, Any]] = None,
**kwargs,
) -> NvidiaPostTrainingJob:
"""
Fine-tunes a model on a dataset.