rosyvs commited on
Commit
f60eaa1
1 Parent(s): 9667a74

add links to isatasr repo to readme

Browse files
Files changed (1) hide show
  1. README.md +9 -2
README.md CHANGED
@@ -4,11 +4,12 @@ language:
4
  library_name: transformers
5
  pipeline_tag: automatic-speech-recognition
6
  ---
7
- Model trained in 8-bit with LoRA
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
@@ -22,4 +23,10 @@ asr_model=prepare_pipeline(
22
  run ASR:
23
  ```
24
  asr_model(audio_path)
25
- ```
 
 
 
 
 
 
 
4
  library_name: transformers
5
  pipeline_tag: automatic-speech-recognition
6
  ---
7
+ Model trained in int8 with LoRA
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
  ```
14
  asr_model=prepare_pipeline(
15
  model_dir='.', # wherever you save the model
 
23
  run ASR:
24
  ```
25
  asr_model(audio_path)
26
+ ```
27
+
28
+ See also:
29
+ https://github.com/rosyvs/isatasr
30
+ Model is on Github at https://github.com/rosyvs/isatasr/tree/main/models/whisat-1.2
31
+ Training script: https://github.com/rosyvs/isatasr/blob/main/train/whisat/tune_hf_whisper.py
32
+ Training hyperparameters: https://github.com/rosyvs/isatasr/blob/main/train/whisat/hparams/redo_for_ICASSP/publicKS_ig_hf_LoRA_int8_largev2.yaml