curry tang
commited on
Commit
•
f15abe4
1
Parent(s):
f6c9b66
update
Browse files
llm.py
CHANGED
@@ -60,9 +60,10 @@ class DeepSeekLLM(BaseLLM):
|
|
60 |
|
61 |
class OpenRouterLLM(BaseLLM):
|
62 |
_support_models = [
|
63 |
-
'openai/gpt-4o-mini', 'anthropic/claude-3.5-sonnet',
|
64 |
-
'
|
65 |
-
'
|
|
|
66 |
'liuhaotian/llava-yi-34b', 'qwen/qwen-110b-chat',
|
67 |
'qwen/qwen-72b-chat', 'google/gemini-pro-1.5',
|
68 |
'cohere/command-r-plus', 'anthropic/claude-3-haiku',
|
|
|
60 |
|
61 |
class OpenRouterLLM(BaseLLM):
|
62 |
_support_models = [
|
63 |
+
'openai/gpt-4o-mini', 'anthropic/claude-3.5-sonnet',
|
64 |
+
'mistralai/mistral-large', 'meta-llama/llama-3.1-405b-instruct', 'openai/gpt-4o',
|
65 |
+
'nvidia/nemotron-4-340b-instruct', 'deepseek/deepseek-coder', '01-ai/yi-large',
|
66 |
+
'google/gemini-flash-1.5', 'deepseek/deepseek-chat', 'qwen/qwen-2-72b-instruct',
|
67 |
'liuhaotian/llava-yi-34b', 'qwen/qwen-110b-chat',
|
68 |
'qwen/qwen-72b-chat', 'google/gemini-pro-1.5',
|
69 |
'cohere/command-r-plus', 'anthropic/claude-3-haiku',
|