phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-31 07:43:54 +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
parent 0a5ca98198
commit e27af206e9
4 changed files with 37 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

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