Sbnos commited on
Commit
c1da23d
1 Parent(s): 17dad20
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -2,7 +2,7 @@ import streamlit as st
2
  import os
3
  from langchain.vectorstores import Chroma
4
  from langchain.embeddings import HuggingFaceBgeEmbeddings
5
- from langchain.llms import Together
6
  from langchain import hub
7
  from operator import itemgetter
8
  from langchain.schema.runnable import RunnableParallel
@@ -39,7 +39,6 @@ embedding_function = HuggingFaceBgeEmbeddings(
39
  llm = Together(
40
  model="mistralai/Mixtral-8x22B-Instruct-v0.1",
41
  temperature=0.2,
42
- max_tokens=24096,
43
  top_k=12,
44
  together_api_key=os.environ['pilotikval']
45
  )
@@ -48,7 +47,6 @@ llm = Together(
48
  llmc = Together(
49
  model="mistralai/Mixtral-8x22B-Instruct-v0.1",
50
  temperature=0.2,
51
- max_tokens=1024,
52
  top_k=3,
53
  together_api_key=os.environ['pilotikval']
54
  )
 
2
  import os
3
  from langchain.vectorstores import Chroma
4
  from langchain.embeddings import HuggingFaceBgeEmbeddings
5
+ from langchain-together import Together
6
  from langchain import hub
7
  from operator import itemgetter
8
  from langchain.schema.runnable import RunnableParallel
 
39
  llm = Together(
40
  model="mistralai/Mixtral-8x22B-Instruct-v0.1",
41
  temperature=0.2,
 
42
  top_k=12,
43
  together_api_key=os.environ['pilotikval']
44
  )
 
47
  llmc = Together(
48
  model="mistralai/Mixtral-8x22B-Instruct-v0.1",
49
  temperature=0.2,
 
50
  top_k=3,
51
  together_api_key=os.environ['pilotikval']
52
  )