Merge pull request #129 from mattf/use-starcoder2-15b-for-tests

use starcoder2-15b for register_model test
langchain-ai · Jan 10, 2025 · aed0964 · aed0964
2 parents f040d8d + 5adb4f2
commit aed0964
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/libs/ai-endpoints/tests/integration_tests/test_register_model.py b/libs/ai-endpoints/tests/integration_tests/test_register_model.py
@@ -37,8 +37,8 @@
         ),
         (
             NVIDIA,
-            "bigcode/starcoder2-7b",
-            "https://api.nvcf.nvidia.com/v2/nvcf/pexec/functions/dd7b01e7-732d-4da5-8e8d-315f79165a23",
+            "bigcode/starcoder2-15b",
+            "https://api.nvcf.nvidia.com/v2/nvcf/pexec/functions/d9cfe8a2-44df-44a0-ba51-3fc4a202c11c",
         ),
     ],
 )