diff --git a/bin/lib/nim-images.json b/bin/lib/nim-images.json index e88faee250a..f639a24454f 100644 --- a/bin/lib/nim-images.json +++ b/bin/lib/nim-images.json @@ -7,7 +7,7 @@ }, { "name": "nvidia/nemotron-3-nano-30b-a3b", - "image": "nvcr.io/nim/nvidia/nemotron-3-nano-30b-a3b:latest", + "image": "nvcr.io/nim/nvidia/nemotron-3-nano:latest", "minGpuMemoryMB": 8192 }, { diff --git a/test/nim.test.js b/test/nim.test.js index 04f29e86265..cd4cf6cd4c7 100644 --- a/test/nim.test.js +++ b/test/nim.test.js @@ -22,7 +22,7 @@ describe("nim", () => { describe("getImageForModel", () => { it("returns correct image for known model", () => { - expect(nim.getImageForModel("nvidia/nemotron-3-nano-30b-a3b")).toBe("nvcr.io/nim/nvidia/nemotron-3-nano-30b-a3b:latest"); + expect(nim.getImageForModel("nvidia/nemotron-3-nano-30b-a3b")).toBe("nvcr.io/nim/nvidia/nemotron-3-nano:latest"); }); it("returns null for unknown model", () => {