diff --git a/nvidia/txt2kg/assets/frontend/components/llm-selector-compact.tsx b/nvidia/txt2kg/assets/frontend/components/llm-selector-compact.tsx index ffaaeba..43a455a 100644 --- a/nvidia/txt2kg/assets/frontend/components/llm-selector-compact.tsx +++ b/nvidia/txt2kg/assets/frontend/components/llm-selector-compact.tsx @@ -22,12 +22,19 @@ const DEFAULT_MODELS: LLMModel[] = [ description: "Local Ollama model" }, { - id: "nvidia-nemotron", - name: "Nemotron 70B", - model: "nvdev/nvidia/llama-3.1-nemotron-70b-instruct", + id: "nvidia-nemotron-super", + name: "Nemotron Super 49B", + model: "nvidia/llama-3.3-nemotron-super-49b-v1.5", provider: "nvidia", description: "NVIDIA API (requires key)" }, + { + id: "nvidia-nemotron-nano", + name: "Nemotron Nano 9B v2", + model: "nvidia/nvidia-nemotron-nano-9b-v2", + provider: "nvidia", + description: "NVIDIA API - Fast & efficient" + }, ] export function LLMSelectorCompact() { diff --git a/nvidia/txt2kg/assets/frontend/components/model-selector.tsx b/nvidia/txt2kg/assets/frontend/components/model-selector.tsx index d3648d2..8f2ceea 100644 --- a/nvidia/txt2kg/assets/frontend/components/model-selector.tsx +++ b/nvidia/txt2kg/assets/frontend/components/model-selector.tsx @@ -18,10 +18,10 @@ const baseModels = [ }, { id: "nvidia-nemotron-nano", - name: "llama-3.1-nemotron-nano-8b-v1", + name: "NVIDIA Nemotron Nano 9B v2", icon: , - description: "NVIDIA hosted Nemotron Nano 8B model", - model: "nvdev/nvidia/llama-3.1-nemotron-nano-8b-instruct", + description: "NVIDIA hosted Nemotron Nano 9B v2 - Faster and more efficient", + model: "nvidia/nvidia-nemotron-nano-9b-v2", apiKeyName: "NVIDIA_API_KEY", baseURL: "https://integrate.api.nvidia.com/v1", },