소스 검색

Safer asset dir handling (#223)

Can Balioglu 1 년 전
부모
커밋
420f8b9ac6
1개의 변경된 파일5개의 추가작업 그리고 3개의 파일을 삭제
  1. 5 3
      src/seamless_communication/store.py

+ 5 - 3
src/seamless_communication/store.py

@@ -14,18 +14,20 @@ from fairseq2.assets import InProcAssetMetadataProvider, asset_store
 def add_gated_assets(model_dir: Path) -> None:
     asset_store.env_resolvers.append(lambda: "gated")
 
+    model_dir = model_dir.resolve()
+
     gated_metadata = [
         {
             "name": "seamless_expressivity@gated",
-            "checkpoint": f"/{model_dir}/m2m_expressive_unity.pt",
+            "checkpoint": model_dir.joinpath("m2m_expressive_unity.pt"),
         },
         {
             "name": "vocoder_pretssel@gated",
-            "checkpoint": f"/{model_dir}/pretssel_melhifigan_wm.pt",
+            "checkpoint": model_dir.joinpath("pretssel_melhifigan_wm.pt"),
         },
         {
             "name": "vocoder_pretssel_16khz@gated",
-            "checkpoint": f"/{model_dir}/pretssel_melhifigan_wm-16khz.pt",
+            "checkpoint": model_dir.joinpath("pretssel_melhifigan_wm-16khz.pt"),
         },
     ]