Spaces:
Running
Running
Commit
·
817c2b8
1
Parent(s):
d082e70
is_chat_model=True
Browse files
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
|