diff --git a/pyproject.toml b/pyproject.toml index 34ac4df..f1373b8 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [project] name = "comfyui-meta-sam-audio" -version = "0.3.0" +version = "0.3.1" description = "Native ComfyUI nodes for local Meta SAM-Audio sound separation" readme = "README.md" requires-python = ">=3.11" diff --git a/sam_audio_comfy/runtime.py b/sam_audio_comfy/runtime.py index e5a45c4..db68dcf 100644 --- a/sam_audio_comfy/runtime.py +++ b/sam_audio_comfy/runtime.py @@ -94,7 +94,7 @@ MIRROR_MODELS = { # snapshots once, then pass local paths upstream to prevent repeated Hub checks. SUPPORT_MODELS = { "t5-base": SupportModelSpec( - "t5-base", + "google-t5/t5-base", "a9723ea7f1b39c1eae772870f3b547bf6ef7e6c1", ("config.json", "model.safetensors", "spiece.model", "tokenizer.json"), ), diff --git a/tests/test_runtime.py b/tests/test_runtime.py index 209eaf5..968e143 100644 --- a/tests/test_runtime.py +++ b/tests/test_runtime.py @@ -248,9 +248,10 @@ def test_model_kwargs_resolve_t5_but_defer_span_predictor(tmp_path, monkeypatch) kwargs = runtime._sam_audio_model_kwargs(tmp_path) assert kwargs["text_encoder"] == { - "name": "/cache/t5-base", + "name": "/cache/google-t5--t5-base", "dim": 768, } + assert runtime.SUPPORT_MODELS["t5-base"].repo_id == "google-t5/t5-base" assert kwargs["span_predictor"] is None assert runtime._configured_span_predictor(tmp_path) == "pe-a-frame-large" assert kwargs["text_ranker"] is None