dinhquangson commited on
Commit
46d4ef9
1 Parent(s): 8c18188

Update QueryMetadataExtractor.py

Browse files
Files changed (1) hide show
  1. QueryMetadataExtractor.py +1 -1
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="mistral-7b-instruct",
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()