diff --git a/llama_stack/providers/remote/inference/nvidia/nvidia.py b/llama_stack/providers/remote/inference/nvidia/nvidia.py
index 4c731a365..9384928aa 100644
--- a/llama_stack/providers/remote/inference/nvidia/nvidia.py
+++ b/llama_stack/providers/remote/inference/nvidia/nvidia.py
@@ -163,8 +163,8 @@ class NVIDIAInferenceAdapter(Inference, ModelRegistryHelper):
 
         if task_type is not None:
             task_type_options = {
-                EmbeddingTaskType.document: "DOCUMENT",
-                EmbeddingTaskType.query: "QUERY",
+                EmbeddingTaskType.document: "passage",
+                EmbeddingTaskType.query: "query",
             }
             if task_type not in task_type_options:
                 raise ValueError(f"Invalid task_type: {task_type}")