exhaustive "llava" match

OthersideAI · May 19, 2024 · f79a549 · f79a549
1 parent 19a8948
commit f79a549
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/operate/models/apis.py b/operate/models/apis.py
@@ -52,7 +52,7 @@ async def get_next_action(model, messages, objective, session_id):
         return "coming soon"
     if model == "gemini-pro-vision":
         return call_gemini_pro_vision(messages, objective), None
-    if model == "llava" or model == "llava:13b" or "bakllava" or "llava-llama3":
+    if "llava" in model:
         operation = call_ollama_llava(messages, model)
         return operation, None
     if model == "claude-3":