Просмотр исходного кода

Add HF Hub Model card + checkpoint. - Conformer shaw (#288)

* Add HF Hub Model card + checkpoint.

* Add load from HUB URL.
Vaibhav Srivastav 1 год назад
Родитель
Сommit
574019c1f6
2 измененных файлов с 2 добавлено и 2 удалено
  1. 1 1
      README.md
  2. 1 1
      src/seamless_communication/cards/conformer_shaw.yaml

+ 1 - 1
README.md

@@ -161,7 +161,7 @@ Please check out above [section](#seamlessexpressive-models) on how to acquire `
 ### W2v-BERT 2.0 speech encoder
 | Model Name        | #params | checkpoint                                                                                                                                                                                                                                                                                                                                                                 |
 | ----------------- | ------- | ------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------- |
-| W2v-BERT 2.0 | 600M    | [checkpoint](https://dl.fbaipublicfiles.com/seamless/models/conformer_shaw.pt)
+| W2v-BERT 2.0 | 600M    | [🤗 Model card](https://huggingface.co/facebook/conformer-shaw) - [checkpoint](https://huggingface.co/facebook/conformer-shaw/resolve/main/conformer_shaw.pt)
 
 Here's how you should do a foward pass through the speech encoder:
 

+ 1 - 1
src/seamless_communication/cards/conformer_shaw.yaml

@@ -7,4 +7,4 @@
 name: conformer_shaw
 model_type: wav2vec2
 model_arch: conformer_shaw_600m
-checkpoint: "https://dl.fbaipublicfiles.com/seamless/models/conformer_shaw.pt"
+checkpoint: "https://huggingface.co/facebook/conformer-shaw/resolve/main/conformer_shaw.pt"