diff --git a/garak/generators/nim.py b/garak/generators/nim.py index 86ac89b84..0379aab24 100644 --- 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