Support more settings

2024-02-19 09:53:26 -06:00 · 2024-02-19 09:53:26 -06:00 · e55d903613
parent 90d5fbc182
commit e55d903613
1 changed files with 4 additions and 0 deletions
--- a/src/constants.js
+++ b/src/constants.js
@ -181,10 +181,14 @@ const TEXTGEN_TYPES = {

 const INFERMATICAI_KEYS = [
    'model',
+    'prompt',
    'max_tokens',
    'temperature',
+    'top_p',
+    'top_k',
    'repetition_penalty',
    'stream',
+    'stop',
 ];

 // https://docs.together.ai/reference/completions