refactor for examples

2026-08-14 23:30:22 +02:00 · 2025-07-26 16:31:49 -06:00
parent 27dad236ef
commit 5b27d4c2e3
4 changed files with 109 additions and 103 deletions
@@ -34,6 +34,7 @@ class TextGenerationFoundationModel(AbstractFoundationModel):
            "text-generation",
            do_sample=True,
            max_new_tokens=512,
+            max_length=1024,
            model=self.model,
            repetition_penalty=1.1,
            temperature=0.3,