CalvinBackup/llmsecops-research
1
0
Fork 0
mirror of https://github.com/lightbroker/llmsecops-research.git synced 2026-08-16 00:00:29 +02:00
Code Issues Packages Projects Releases Wiki Activity

LLM config output

Browse Source
This commit is contained in:
Adam Wilson
2025-07-27 11:21:12 -06:00
parent a7a6873e73
commit eddacd87fa
7 changed files with 616 additions and 254 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/text_generation/adapters/text_generation_foundation_model.py
+1
View File
@@ -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,