clementsan commited on
Commit
3324680
1 Parent(s): 989bff5

Add new LLM: zephyr-7b-beta

Browse files
Files changed (1) hide show
  1. app.py +8 -7
app.py CHANGED
@@ -26,13 +26,14 @@ import accelerate
26
  llm_name0 = "mistralai/Mixtral-8x7B-Instruct-v0.1"
27
  llm_name1 = "mistralai/Mistral-7B-Instruct-v0.2"
28
  llm_name2 = "mistralai/Mistral-7B-Instruct-v0.1"
29
- llm_name3 = "meta-llama/Llama-2-7b-chat-hf"
30
- llm_name4 = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
31
- llm_name5 = "microsoft/phi-2"
32
- llm_name6 = "mosaicml/mpt-7b-instruct"
33
- llm_name7 = "tiiuae/falcon-7b-instruct"
34
- llm_name8 = "google/flan-t5-xxl"
35
- list_llm = [llm_name0, llm_name1, llm_name2, llm_name3, llm_name4, llm_name5, llm_name6, llm_name7, llm_name8]
 
36
  list_llm_simple = [os.path.basename(llm) for llm in list_llm]
37
 
38
  # Load PDF document and create doc splits
 
26
  llm_name0 = "mistralai/Mixtral-8x7B-Instruct-v0.1"
27
  llm_name1 = "mistralai/Mistral-7B-Instruct-v0.2"
28
  llm_name2 = "mistralai/Mistral-7B-Instruct-v0.1"
29
+ llm_name3 = "mistralai/Mistral-7B-Instruct-v0.1"
30
+ llm_name4 = "HuggingFaceH4/zephyr-7b-beta"
31
+ llm_name5 = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
32
+ llm_name6 = "microsoft/phi-2"
33
+ llm_name7 = "mosaicml/mpt-7b-instruct"
34
+ llm_name8 = "tiiuae/falcon-7b-instruct"
35
+ llm_name9 = "google/flan-t5-xxl"
36
+ list_llm = [llm_name0, llm_name1, llm_name2, llm_name3, llm_name4, llm_name5, llm_name6, llm_name7, llm_name8, llm_name9]
37
  list_llm_simple = [os.path.basename(llm) for llm in list_llm]
38
 
39
  # Load PDF document and create doc splits