HarounElleuch commited on
Commit
e233eb3
·
verified ·
1 Parent(s): 7c6e64f

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. README.md +2 -2
  2. hyperparams.yaml +3 -1
README.md CHANGED
@@ -42,9 +42,9 @@ pip install git+https://github.com/speechbrain/speechbrain.git@develop
42
  from inference.classifier_attention_pooling import WhisperDialectClassifier
43
 
44
  dialect_id = WhisperDialectClassifier.from_hparams(
45
- source="",
46
  hparams_file="hyperparams.yaml",
47
- savedir="pretrained_DID/tmp",
48
  run_opts={"device": "cuda"} # If using a GPU (recommended).
49
  )
50
 
 
42
  from inference.classifier_attention_pooling import WhisperDialectClassifier
43
 
44
  dialect_id = WhisperDialectClassifier.from_hparams(
45
+ source="Elyadata/ADI-whisper-ADI17",
46
  hparams_file="hyperparams.yaml",
47
+ savedir="pretrained_DID",
48
  run_opts={"device": "cuda"} # If using a GPU (recommended).
49
  )
50
 
hyperparams.yaml CHANGED
@@ -11,11 +11,13 @@ whisper_hub: openai/whisper-large-v3
11
  n_languages: 17
12
  features_dim: 1280
13
 
 
 
14
  whisper: !new:speechbrain.lobes.models.huggingface_transformers.whisper.Whisper
15
  source: !ref <whisper_hub>
16
  encoder_only: True
17
  freeze_encoder: False
18
- save_path: !ref <whisper_hub>
19
 
20
  attention_pooling: !new:speechbrain.nnet.pooling.AttentionPooling
21
  input_dim: !ref <features_dim>
 
11
  n_languages: 17
12
  features_dim: 1280
13
 
14
+ # For newer versions of SpeechBrain (dev branch), replace with:
15
+ # whisper: !new:speechbrain.integrations.huggingface.whisper.Whisper
16
  whisper: !new:speechbrain.lobes.models.huggingface_transformers.whisper.Whisper
17
  source: !ref <whisper_hub>
18
  encoder_only: True
19
  freeze_encoder: False
20
+ save_path: whisper_hf
21
 
22
  attention_pooling: !new:speechbrain.nnet.pooling.AttentionPooling
23
  input_dim: !ref <features_dim>