Skip to content

Commit

Permalink
Merge branch 'development' into falcon
Browse files Browse the repository at this point in the history
  • Loading branch information
MithilShah authored Aug 14, 2023
2 parents 749afbd + 3889639 commit 1ed398b
Show file tree
Hide file tree
Showing 9 changed files with 1,445 additions and 5 deletions.
5 changes: 4 additions & 1 deletion .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -82,4 +82,7 @@ dmypy.json
.DS_Store

# vs code
.vscode
.vscode

# venv files
env*
6 changes: 5 additions & 1 deletion kendra_retriever_samples/app.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@
import kendra_chat_flan_xxl as flanxxl
import kendra_chat_open_ai as openai
import kendra_chat_falcon_40b as falcon40b

import kendra_chat_llama_2 as llama2

USER_ICON = "images/user-icon.png"
AI_ICON = "images/ai-icon.png"
Expand All @@ -18,6 +18,7 @@
'flanxl': 'Flan XL',
'flanxxl': 'Flan XXL',
'falcon40b': 'Falcon 40B'
'llama2' : 'Llama 2'
}

# Check if the user ID is already stored in the session state
Expand Down Expand Up @@ -47,6 +48,9 @@
elif (sys.argv[1] == 'falcon40b'):
st.session_state['llm_app'] = falcon40b
st.session_state['llm_chain'] = falcon40b.build_chain()
elif (sys.argv[1] == 'llama2'):
st.session_state['llm_app'] = llama2
st.session_state['llm_chain'] = llama2.build_chain()
else:
raise Exception("Unsupported LLM: ", sys.argv[1])
else:
Expand Down
Loading

0 comments on commit 1ed398b

Please sign in to comment.