adjust model config args

This commit is contained in:
Adam Wilson
2025-06-24 14:23:08 -06:00
parent f867a4fdc9
commit b1cd2a02b8

View File

@@ -47,16 +47,22 @@ class TextGenerationFoundationModel:
# Create the text generation pipeline
pipe = pipeline(
"text-generation",
do_sample=True,
max_new_tokens=512,
model=model,
tokenizer=tokenizer,
max_new_tokens=256,
temperature=0.7,
top_p=0.9,
repetition_penalty=1.1,
temperature=0.3,
tokenizer=tokenizer,
use_fast=True,
do_sample=True
pad_token_id=tokenizer.eos_token_id,
eos_token_id=tokenizer.eos_token_id,
)
# Create the LangChain LLM
return HuggingFacePipeline(pipeline=pipe)
return HuggingFacePipeline(
pipeline=pipe,
pipeline_kwargs={
"return_full_text": False,
"stop_sequence": ["<|end|>", "<|user|>", "</s>"]
})