mirror of
https://github.com/kvcache-ai/ktransformers.git
synced 2025-09-07 21:19:51 +00:00
fix typo for top_p
This commit is contained in:
parent
69382e58f9
commit
3aa0cfc29d
1 changed files with 1 additions and 1 deletions
|
@ -35,7 +35,7 @@ class KTransformersInterface(TransformersInterface):
|
||||||
generation_config = GenerationConfig(
|
generation_config = GenerationConfig(
|
||||||
max_length=args.max_new_tokens,
|
max_length=args.max_new_tokens,
|
||||||
temperature=args.temperature,
|
temperature=args.temperature,
|
||||||
top_p=args.temperature,
|
top_p=args.top_p,
|
||||||
do_sample=True
|
do_sample=True
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
Loading…
Add table
Reference in a new issue