Fix ctransformers seed

This commit is contained in:
oobabooga 2023-08-13 05:48:53 -07:00
parent 919a3cf9d0
commit f6db2c78d1

View File

@ -56,7 +56,7 @@ class CtransformersModel:
top_k=state['top_k'], top_k=state['top_k'],
repetition_penalty=state['repetition_penalty'], repetition_penalty=state['repetition_penalty'],
last_n_tokens=state['repetition_penalty_range'], last_n_tokens=state['repetition_penalty_range'],
seed=state['seed'] seed=int(state['seed'])
) )
output = "" output = ""