Home Explore Help
Register Sign In
github
/
seamless_communication
mirror of https://github.com/facebookresearch/seamless_communication.git
2
0
Fork 0
Files Issues 0 Wiki
Browse Source

Fix medium model loading (#144)

Ning 1 year ago
parent
a1b5d918eb
commit
fad6308372
1 changed files with 1 additions and 1 deletions
Split View Show Diff Stats
  1. 1 1
      src/seamless_communication/models/unity/builder.py

+ 1 - 1
src/seamless_communication/models/unity/builder.py
View File 

@@ -139,7 +139,7 @@ def _medium() -> UnitYConfig:
 
 
     mt_model_config: NllbConfig = nllb_archs.get_config("dense_600m")
 
 
-    mt_model_config.vocab_info.size = 256102  # NLLB-100
 
+    mt_model_config.vocab_info.size = 256206  # NLLB-200
 
 
     t2u_config = unity_t2u_archs.get_config("medium")