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