From 7661255b2a3de6d04270588409ae2dfae6dfb5c6 Mon Sep 17 00:00:00 2001 From: Paolo Massa Date: Wed, 23 Oct 2024 13:17:44 +0200 Subject: [PATCH] Update README.md In docs, TextGeneratorConfig is used but does not exist. The correct name is TextGenerationConfig --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 23908a3..221f0dd 100644 --- a/README.md +++ b/README.md @@ -32,7 +32,7 @@ messages = [ - **Good Utils (e.g., Caching etc)**: E.g. being able to use caching for faster responses. General policy is that cache is used if config (including messages) is the same. If you want to force a new response, set `use_cache=False` in the `generate` call. ```python -response = gen.generate(messages=messages, config=TextGeneratorConfig(n=1, use_cache=True)) +response = gen.generate(messages=messages, config=TextGenerationConfig(n=1, use_cache=True)) ``` Output looks like