diff --git a/llama_stack/providers/remote/post_training/nvidia/post_training.py b/llama_stack/providers/remote/post_training/nvidia/post_training.py index bacfdba0b..e14fcf0cc 100644 --- a/llama_stack/providers/remote/post_training/nvidia/post_training.py +++ b/llama_stack/providers/remote/post_training/nvidia/post_training.py @@ -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.