fix bug tom found

Learning-and-Intelligent-Systems · Nov 4, 2024 · cdc30e7 · cdc30e7
1 parent 337e789
commit cdc30e7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/predicators/pretrained_model_interface.py b/predicators/pretrained_model_interface.py
@@ -245,7 +245,8 @@ def _sample_completions(
         responses = [
             self.call_openai_api(messages,
                                  model=self._model_name,
-                                 temperature=temperature)
+                                 temperature=temperature,
+                                 max_tokens=self._max_tokens)
             for _ in range(num_completions)
         ]
         return responses