phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-31 05:50:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add model mapping

Browse source
This commit is contained in:
Ubuntu 2025-03-12 14:32:08 +00:00 • committed by raspawar
parent bd9b6a6e00
commit d7ead08cb9
4 changed files with 37 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

5
tests/unit/providers/nvidia/__init__.py Normal file
View file

@ -0,0 +1,5 @@
# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
#
# This source code is licensed under the terms described in the LICENSE file in
# the root directory of this source tree.

1
tests/unit/providers/nvidia/test_supervised_fine_tuning.py
View file

@ -22,6 +22,7 @@ from llama_stack.distribution.library_client import LlamaStackAsLibraryClient
class TestNvidiaPostTraining(unittest.TestCase):
# ToDo: add tests for env variables, models supported.
def setUp(self):
os.environ["NVIDIA_BASE_URL"] = "http://nemo.test" # needed for llm inference
os.environ["NVIDIA_CUSTOMIZER_URL"] = "http://nemo.test" # needed for nemo customizer