소스 검색

update model names

Maha Elbayad 2 년 전
부모
커밋
7722b789b2
2개의 변경된 파일100개의 추가작업 그리고 0개의 파일을 삭제
  1. 50 0
      src/seamless_communication/assets/cards/seamlessM4T_large.yaml
  2. 50 0
      src/seamless_communication/assets/cards/seamlessM4T_medium.yaml

+ 50 - 0
src/seamless_communication/assets/cards/seamlessM4T_large.yaml

@@ -0,0 +1,50 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the BSD-style license found in the
+# LICENSE file in the root directory of this source tree.
+
+name: seamlessM4T_large
+base: unity_nllb-100
+model_arch: base
+checkpoint: "https://dl.fbaipublicfiles.com/seamless_aug/models/large_unity/multitask_unity_large.pt"
+num_units: 10000
+unit_langs:
+  - arb
+  - ben
+  - cat
+  - ces
+  - cmn
+  - cym
+  - dan
+  - deu
+  - eng
+  - est
+  - fin
+  - fra
+  - hin
+  - ind
+  - ita
+  - jpn
+  - kan
+  - kor
+  - mlt
+  - nld
+  - pes
+  - pol
+  - por
+  - ron
+  - rus
+  - slk
+  - spa
+  - swe
+  - swh
+  - tam
+  - tel
+  - tgl
+  - tha
+  - tur
+  - ukr
+  - urd
+  - uzn
+  - vie

+ 50 - 0
src/seamless_communication/assets/cards/seamlessM4T_medium.yaml

@@ -0,0 +1,50 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the BSD-style license found in the
+# LICENSE file in the root directory of this source tree.
+
+name: seamlessM4T_medium 
+base: unity_nllb-200
+model_arch: medium
+checkpoint: "https://dl.fbaipublicfiles.com/seamless_aug/models/medium_unity/multitask_unity_medium.pt"
+num_units: 10000
+unit_langs:
+  - arb
+  - ben
+  - cat
+  - ces
+  - cmn
+  - cym
+  - dan
+  - deu
+  - eng
+  - est
+  - fin
+  - fra
+  - hin
+  - ind
+  - ita
+  - jpn
+  - kan
+  - kor
+  - mlt
+  - nld
+  - pes
+  - pol
+  - por
+  - ron
+  - rus
+  - slk
+  - spa
+  - swe
+  - swh
+  - tam
+  - tel
+  - tgl
+  - tha
+  - tur
+  - ukr
+  - urd
+  - uzn
+  - vie