Add Model: falcon-180b-chat

ramon-victor · Sep 16, 2023 · 4f616c1 · 4f616c1
1 parent 7ab6736
commit 4f616c1
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 3 deletions.
diff --git a/client/html/index.html b/client/html/index.html
@@ -116,9 +116,10 @@
  <option value="gpt-4-0314">GPT-4-0314</option>
  <option value="gpt-4-32k">GPT-4-32k</option>
  </optgroup>
- <optgroup label="LLAMA">
+ <optgroup label="OPEN LLM">
  <option value="oasst-sft-6-llama-30b">LLaMa-30B-sft-6</option>
  <option value="llama-2-70b-chat">LLaMa-2-70B-chat</option>
+ <option value="falcon-180b-chat">Falcon-180b-chat</option>
  </optgroup>
  <optgroup label="CLAUDE">
  <option value="claude-instant">Claude-instant</option>

diff --git a/g4f/Provider/Providers/Naga.py b/g4f/Provider/Providers/Naga.py
@@ -17,8 +17,9 @@
  'gpt-4',
  'gpt-4-0314',
  'gpt-4-32k',
- 'llama-2-70b-chat',
  'oasst-sft-6-llama-30b',
+ 'llama-2-70b-chat',
+ 'falcon-180b-chat',
  'claude-instant',
  'claude-2',
  'claude-2-100k'

diff --git a/g4f/models.py b/g4f/models.py
@@ -97,6 +97,11 @@ class oasst_sft_6_llama_30b:
  base_provider: str = 'huggingface-chat'
  best_provider: Provider.Provider = Provider.Naga
 
+ class falcon_180b_chat:
+ name: str = 'falcon-180b-chat'
+ base_provider: str = 'huggingface-chat'
+ best_provider: Provider.Provider = Provider.Naga
+
 
 
 class ModelUtils:
@@ -117,8 +122,9 @@ class ModelUtils:
  'claude-2': Model.claude_2,
  'claude-2-100k': Model.claude_2_100k,
 
+ 'oasst-sft-6-llama-30b': Model.oasst_sft_6_llama_30b,
  'llama-2-7b-chat': Model.llama_2_7b_chat,
  'llama-2-13b-chat': Model.llama_2_13b_chat,
  'llama-2-70b-chat': Model.llama_2_70b_chat,
- 'oasst-sft-6-llama-30b': Model.oasst_sft_6_llama_30b,
+ 'falcon-180b-chat': Model.falcon_180b_chat
  }