curry tang commited on
Commit
007f9f4
·
1 Parent(s): d9cc5c0
Files changed (1) hide show
  1. llm.py +2 -2
llm.py CHANGED
@@ -60,9 +60,9 @@ class DeepSeekLLM(BaseLLM):
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',
 
60
 
61
  class OpenRouterLLM(BaseLLM):
62
  _support_models = [
63
+ 'openai/gpt-4o-mini', 'anthropic/claude-3.5-sonnet', 'google/gemini-pro-1.5-exp',
64
  'mistralai/mistral-large', 'meta-llama/llama-3.1-405b-instruct', 'openai/gpt-4o',
65
+ 'nvidia/nemotron-4-340b-instruct', 'deepseek/deepseek-coder', 'google/gemma-2-27b-it',
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',