Fix missing llm key

argilla-io · Apr 10, 2024 · 4361fbd · 4361fbd
1 parent 0d3544f
commit 4361fbd
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 6 deletions.
diff --git a/README.md b/README.md
@@ -128,9 +128,11 @@ if __name__ == "__main__":
                 "split": "test",
             },
             "generate_with_openai": {
-                "generation_kwargs": {
-                    "temperature": 0.7,
-                    "max_new_tokens": 512,
+                "llm": {
+                    "generation_kwargs": {
+                        "temperature": 0.7,
+                        "max_new_tokens": 512,
+                    }
                 }
             },
             "to_argilla": {

diff --git a/docs/index.md b/docs/index.md
@@ -66,9 +66,11 @@ if __name__ == "__main__":
                 "split": "test",
             },
             "generate_with_openai": {
-                "generation_kwargs": {
-                    "temperature": 0.7,
-                    "max_new_tokens": 512,
+                "llm": {
+                    "generation_kwargs": {
+                        "temperature": 0.7,
+                        "max_new_tokens": 512,
+                    }
                 }
             },
             "to_argilla": {