add description for model options
Browse files
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':
|
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 |
)
|