mirror of
https://github.com/lightbroker/llmsecops-research.git
synced 2026-09-30 13:29:39 +02:00
refactor for examples
This commit is contained in:
1 parent
27dad236ef
commit
5b27d4c2e3
4 files changed
+109
-103
No files matched your search
@@ -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,
|
||||
|
||||
Reference in new issue
Block a user