aldohenrique commited on
Commit
f8cb96f
·
verified ·
1 Parent(s): bdf9792

Update ai_logic.py

Browse files
Files changed (1) hide show
  1. ai_logic.py +3 -1
ai_logic.py CHANGED
@@ -20,7 +20,9 @@ if not HF_TOKEN:
20
 
21
  MODELS = {
22
  "Phi-3 Mini": "microsoft/Phi-3-mini-4k-instruct",
23
- "Mistral 7B": "mistralai/Mistral-7B-Instruct-v0.3"
 
 
24
  }
25
  DEFAULT_MODEL = "Phi-3 Mini"
26
 
 
20
 
21
  MODELS = {
22
  "Phi-3 Mini": "microsoft/Phi-3-mini-4k-instruct",
23
+ "Mistral 7B": "mistralai/Mistral-7B-Instruct-v0.3",
24
+ "Deepseek (chat) 7B":"deepseek-ai/deepseek-vl-7b-chat",
25
+ "Gemma (Google) 7B":"google/gemma-7b-it"
26
  }
27
  DEFAULT_MODEL = "Phi-3 Mini"
28