datacipen commited on
Commit
3a12ccd
1 Parent(s): 21f76f2

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -922,7 +922,7 @@ async def API_connexion(romeListArray):
922
  @cl.step(type="llm")
923
  async def IA():
924
  os.environ['HUGGINGFACEHUB_API_TOKEN'] = os.environ['HUGGINGFACEHUB_API_TOKEN']
925
- repo_id = "mistralai/Mixtral-8x7B-Instruct-v0.1"
926
 
927
  llm = HuggingFaceEndpoint(
928
  repo_id=repo_id, max_new_tokens=8000, temperature=0.5, streaming=True
@@ -983,7 +983,7 @@ async def setup_agent(settings):
983
  contextChat = "Il n'y a pas de contexte."
984
 
985
  os.environ['HUGGINGFACEHUB_API_TOKEN'] = os.environ['HUGGINGFACEHUB_API_TOKEN']
986
- repo_id = "mistralai/Mixtral-8x7B-Instruct-v0.1"
987
 
988
  model = HuggingFaceEndpoint(
989
  repo_id=repo_id, max_new_tokens=8000, temperature=0.5, streaming=True
 
922
  @cl.step(type="llm")
923
  async def IA():
924
  os.environ['HUGGINGFACEHUB_API_TOKEN'] = os.environ['HUGGINGFACEHUB_API_TOKEN']
925
+ repo_id = "mistralai/Mixtral-8x22B-Instruct-v0.1"
926
 
927
  llm = HuggingFaceEndpoint(
928
  repo_id=repo_id, max_new_tokens=8000, temperature=0.5, streaming=True
 
983
  contextChat = "Il n'y a pas de contexte."
984
 
985
  os.environ['HUGGINGFACEHUB_API_TOKEN'] = os.environ['HUGGINGFACEHUB_API_TOKEN']
986
+ repo_id = "mistralai/Mixtral-8x22B-Instruct-v0.1"
987
 
988
  model = HuggingFaceEndpoint(
989
  repo_id=repo_id, max_new_tokens=8000, temperature=0.5, streaming=True