Update main.py
Browse files
main.py
CHANGED
@@ -50,7 +50,7 @@ def LLModel():
|
|
50 |
os.environ['HUGGINGFACEHUB_API_TOKEN'] = os.environ['HUGGINGFACEHUB_API_TOKEN']
|
51 |
repo_id = "mistralai/Mixtral-8x7B-Instruct-v0.1"
|
52 |
llm = HuggingFaceEndpoint(
|
53 |
-
repo_id=repo_id, max_new_tokens=
|
54 |
)
|
55 |
add_llm_provider(
|
56 |
LangchainGenericProvider(
|
|
|
50 |
os.environ['HUGGINGFACEHUB_API_TOKEN'] = os.environ['HUGGINGFACEHUB_API_TOKEN']
|
51 |
repo_id = "mistralai/Mixtral-8x7B-Instruct-v0.1"
|
52 |
llm = HuggingFaceEndpoint(
|
53 |
+
repo_id=repo_id, max_new_tokens=5300, temperature=1.0, task="text2text-generation", streaming=True
|
54 |
)
|
55 |
add_llm_provider(
|
56 |
LangchainGenericProvider(
|