k4d3 commited on
Commit
f300b48
1 Parent(s): 3830d24

add repetition penalty for generating captions

Browse files
Files changed (1) hide show
  1. joy +1 -0
joy CHANGED
@@ -470,6 +470,7 @@ class JoyCaptionModel:
470
  max_new_tokens=300,
471
  do_sample=True,
472
  suppress_tokens=None,
 
473
  )
474
  return generate_ids
475
 
 
470
  max_new_tokens=300,
471
  do_sample=True,
472
  suppress_tokens=None,
473
+ repetition_penalty=1.2,
474
  )
475
  return generate_ids
476