curry tang commited on
Commit
f15abe4
1 Parent(s): f6c9b66
Files changed (1) hide show
  1. llm.py +4 -3
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', 'openai/gpt-4o',
64
- 'nvidia/nemotron-4-340b-instruct', 'deepseek/deepseek-coder',
65
- 'google/gemini-flash-1.5', 'deepseek/deepseek-chat',
 
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',