Skip to content

Commit e960dfd

Browse files
committed
Fix the temperature argument to accept values passed via the cmd line.
1 parent 10a8842 commit e960dfd

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

eval/humaneval.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -119,7 +119,7 @@
119119

120120
gen_settings = ExLlamaV2Sampler.Settings(
121121
token_repetition_penalty = 1.0,
122-
temperature = 0.6,
122+
temperature = args.temperature if args.temperature is not None else 0.6,
123123
top_k = 50,
124124
top_p = 0.6
125125
)

0 commit comments

Comments
 (0)