Update main.py
Browse files
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=
|
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 |
|