소스 검색

Change function name: _build_model_config -> _build_seamless_nano_model_config

Ruslan Mavlyutov 1 년 전
부모
커밋
4dc9601538
1개의 변경된 파일3개의 추가작업 그리고 3개의 파일을 삭제
  1. 3 3
      src/seamless_communication/models/unity/builder.py

+ 3 - 3
src/seamless_communication/models/unity/builder.py

@@ -224,7 +224,7 @@ def _expressivity_v2() -> UnitYConfig:
     )
 
 
-def _build_model_config(
+def _build_seamless_nano_model_config(
     model_embed_dim: int,
     ffn_emb_dim_mult: int,
     feature_stride: int,
@@ -334,7 +334,7 @@ def _build_model_config(
 
 @unity_arch("seamless_micro")
 def _seamless_micro() -> UnitYConfig:
-    return _build_model_config(
+    return _build_seamless_nano_model_config(
         model_embed_dim=512,
         ffn_emb_dim_mult=8,
         feature_stride=4,
@@ -346,7 +346,7 @@ def _seamless_micro() -> UnitYConfig:
 
 @unity_arch("seamless_nano")
 def _seamless_nano() -> UnitYConfig:
-    return _build_model_config(
+    return _build_seamless_nano_model_config(
         model_embed_dim=256,
         ffn_emb_dim_mult=8,
         feature_stride=4,