Upload folder using huggingface_hub
Browse files- README.md +1 -1
- 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:
|
| 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>
|