Tristan107 commited on
Commit
817c2b8
·
1 Parent(s): d082e70

is_chat_model=True

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ st.title(PAGE_TITLE)
33
 
34
  #custom_llm = OpenAILike(model=LLM_MODEL, api_base=API_BASE_URL, api_key=st.secrets["openai_key"], max_tokens=MAX_NB_TOKENS_IN_RESPONSE, temperature=TEMPERATURE)
35
  #With local Mistral API key :
36
- custom_llm = OpenAILike(model=LLM_MODEL, api_base=API_BASE_URL, api_key=st.secrets["local_mistral_api_key"], max_tokens=MAX_NB_TOKENS_IN_RESPONSE, temperature=TEMPERATURE)
37
 
38
  Settings.embed_model = HuggingFaceEmbedding(model_name=EMBEDDING_MODEL)
39
  Settings.llm=custom_llm
 
33
 
34
  #custom_llm = OpenAILike(model=LLM_MODEL, api_base=API_BASE_URL, api_key=st.secrets["openai_key"], max_tokens=MAX_NB_TOKENS_IN_RESPONSE, temperature=TEMPERATURE)
35
  #With local Mistral API key :
36
+ custom_llm = OpenAILike(model=LLM_MODEL, api_base=API_BASE_URL, api_key=st.secrets["local_mistral_api_key"], is_chat_model=True, max_tokens=MAX_NB_TOKENS_IN_RESPONSE, temperature=TEMPERATURE)
37
 
38
  Settings.embed_model = HuggingFaceEmbedding(model_name=EMBEDDING_MODEL)
39
  Settings.llm=custom_llm