mirror of
https://github.com/lightbroker/llmsecops-research.git
synced 2026-09-30 13:29:39 +02:00
LLM config output
This commit is contained in:
1 parent
a7a6873e73
commit
eddacd87fa
7 files changed
+616
-254
No files matched your search
@@ -43,6 +43,7 @@ class TextGenerationFoundationModel(AbstractFoundationModel):
|
||||
pad_token_id=self.tokenizer.eos_token_id,
|
||||
eos_token_id=self.tokenizer.eos_token_id
|
||||
)
|
||||
# TODO: fix Both `max_new_tokens` (=512) and `max_length`(=1024) seem to have been set. `max_new_tokens` will take precedence. Please refer to the documentation for more information. (https://huggingface.co/docs/transformers/main/en/main_classes/text_generation)
|
||||
|
||||
return HuggingFacePipeline(
|
||||
pipeline=pipe,
|
||||
|
||||
Reference in new issue
Block a user