HarounElleuch commited on
Commit
51169e9
·
verified ·
1 Parent(s): 7080a56

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. README.md +1 -1
  2. hyperparams.yaml +3 -1
README.md CHANGED
@@ -42,7 +42,7 @@ 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).
 
42
  from inference.classifier_attention_pooling import WhisperDialectClassifier
43
 
44
  dialect_id = WhisperDialectClassifier.from_hparams(
45
+ source="Elyadata/ADI-whisper-ADI20",
46
  hparams_file="hyperparams.yaml",
47
  savedir="pretrained_DID/tmp",
48
  run_opts={"device": "cuda"} # If using a GPU (recommended).
hyperparams.yaml CHANGED
@@ -11,11 +11,13 @@ whisper_hub: openai/whisper-large-v3
11
  n_languages: 20
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: 20
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>