clementsan commited on
Commit
62fa94f
1 Parent(s): b1ec9ac

Add new LLM model microsoft_phi-2

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -23,10 +23,11 @@ default_persist_directory = './chroma_HF/'
23
  llm_name1 = "mistralai/Mistral-7B-Instruct-v0.2"
24
  llm_name2 = "mistralai/Mistral-7B-Instruct-v0.1"
25
  llm_name3 = "meta-llama/Llama-2-7b-chat-hf"
26
- llm_name4 = "mosaicml/mpt-7b-instruct"
27
- llm_name5 = "tiiuae/falcon-7b-instruct"
28
- llm_name6 = "google/flan-t5-xxl"
29
- list_llm = [llm_name1, llm_name2, llm_name3, llm_name4, llm_name5, llm_name6]
 
30
  list_llm_simple = [os.path.basename(llm) for llm in list_llm]
31
 
32
  # Load PDF document and create doc splits
 
23
  llm_name1 = "mistralai/Mistral-7B-Instruct-v0.2"
24
  llm_name2 = "mistralai/Mistral-7B-Instruct-v0.1"
25
  llm_name3 = "meta-llama/Llama-2-7b-chat-hf"
26
+ llm_name4 = "microsoft/phi-2"
27
+ llm_name5 = "mosaicml/mpt-7b-instruct"
28
+ llm_name6 = "tiiuae/falcon-7b-instruct"
29
+ llm_name7 = "google/flan-t5-xxl"
30
+ list_llm = [llm_name1, llm_name2, llm_name3, llm_name4, llm_name5, llm_name6, llm_name7]
31
  list_llm_simple = [os.path.basename(llm) for llm in list_llm]
32
 
33
  # Load PDF document and create doc splits