Fix ctransformers seed
This commit is contained in:
parent
919a3cf9d0
commit
f6db2c78d1
1 changed files with 1 additions and 1 deletions
|
@ -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 = ""
|
||||||
|
|
Loading…
Add table
Reference in a new issue