datacipen commited on
Commit
8a9911a
·
verified ·
1 Parent(s): fda971e

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -123,7 +123,7 @@ def Connexion_Mistral():
123
  #return Mistral(api_key=os.environ["GITHUB_TOKEN"], server_url=endpoint)
124
  repo_id = "mistralai/Mistral-7B-Instruct-v0.2"
125
  llm = HuggingFaceEndpoint(
126
- repo_id=repo_id, max_new_tokens=5300, temperature=0.1, task="text2text-generation", streaming=True
127
  )
128
  return llm
129
 
 
123
  #return Mistral(api_key=os.environ["GITHUB_TOKEN"], server_url=endpoint)
124
  repo_id = "mistralai/Mistral-7B-Instruct-v0.2"
125
  llm = HuggingFaceEndpoint(
126
+ repo_id=repo_id, max_new_tokens=3000, temperature=0.1, task="text2text-generation", streaming=True
127
  )
128
  return llm
129