rosyvs commited on
Commit
4db98ee
1 Parent(s): d0dded1

add description for model options

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -8,13 +8,13 @@ Model trained in 8-bit with LoRA
8
 
9
  Usage:
10
 
11
- prepare pipeline:
12
  ```
13
  asr_model=prepare_pipeline(
14
  model_dir='.', # wherever you save the model
15
  generate_opts={'max_new_tokens':112,
16
  'num_beams':1,
17
- 'repetition_penalty':r1,
18
  'do_sample':False
19
  }
20
  )
 
8
 
9
  Usage:
10
 
11
+ prepare pipeline, setting to default generate_opts will give you (deterministic) greedy decoding with up to 112 tokens generated, no repetition penalty:
12
  ```
13
  asr_model=prepare_pipeline(
14
  model_dir='.', # wherever you save the model
15
  generate_opts={'max_new_tokens':112,
16
  'num_beams':1,
17
+ 'repetition_penalty':1,
18
  'do_sample':False
19
  }
20
  )