Spaces:
Sleeping
Sleeping
dinhquangson
commited on
Commit
•
46d4ef9
1
Parent(s):
8c18188
Update QueryMetadataExtractor.py
Browse files
QueryMetadataExtractor.py
CHANGED
@@ -37,7 +37,7 @@ class QueryMetadataExtractor:
|
|
37 |
generator = OpenAIGenerator(
|
38 |
api_key=Secret.from_env_var("OCTOAI_TOKEN"),
|
39 |
api_base_url="https://text.octoai.run/v1",
|
40 |
-
model="
|
41 |
generation_kwargs = {"max_tokens": 512}
|
42 |
)
|
43 |
self.pipeline = Pipeline()
|
|
|
37 |
generator = OpenAIGenerator(
|
38 |
api_key=Secret.from_env_var("OCTOAI_TOKEN"),
|
39 |
api_base_url="https://text.octoai.run/v1",
|
40 |
+
model="mixtral-8x22b-instruct",
|
41 |
generation_kwargs = {"max_tokens": 512}
|
42 |
)
|
43 |
self.pipeline = Pipeline()
|