Commit 3aa0cfc2 authored by Wix Woo's avatar Wix Woo
Browse files

fix typo for top_p

parent 69382e58
...@@ -35,7 +35,7 @@ class KTransformersInterface(TransformersInterface): ...@@ -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
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment