Inicio Explorar Axuda
Rexistro Iniciar sesión
github
/
seamless_communication
réplica de https://github.com/facebookresearch/seamless_communication.git
2
0
Fork 0
Ficheiros Incidencias 0 Wiki
Explorar o código

Fix medium model loading (#144)

Ning hai 1 ano
pai
a1b5d918eb
achega
fad6308372
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
Dividir vista Mostrar estatísticas de Diff
  1. 1 1
      src/seamless_communication/models/unity/builder.py

+ 1 - 1
src/seamless_communication/models/unity/builder.py
Ver ficheiro 

@@ -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")