rename generator

NVIDIA · Oct 29, 2024 · a24577a · a24577a
1 parent dcbdc83
commit a24577a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/garak/generators/nim.py b/garak/generators/nim.py
@@ -137,7 +137,7 @@ def _load_client(self):
         self.generator = self.client.completions
 
 
-class NIMVision(NVOpenAIChat):
+class Vision(NVOpenAIChat):
     """Wrapper for text+image to text NIMs. Expects NIM_API_KEY environment variable.
 
     Following generators.huggingface.LLaVa, expects prompts to be a dict with keys