Remove local path option from model loader

Models always download to ComfyUI/models/omnivoice/ via HuggingFace.
Local path added unnecessary complexity; users who want a custom path
can symlink into the models directory.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

tests/test_loader.py

@@ -1,6 +1,5 @@
 # tests/test_loader.py
 from unittest.mock import patch, MagicMock
-import torch
 import pytest
 from nodes.loader import OmniVoiceModelLoader
 
@@ -8,52 +7,38 @@ from nodes.loader import OmniVoiceModelLoader
 def test_input_types_structure():
     inputs = OmniVoiceModelLoader.INPUT_TYPES()
     required = inputs["required"]
-    assert "model_source" in required
     assert "device" in required
     assert "dtype" in required
-    optional = inputs.get("optional", {})
-    assert "local_path" in optional
+    assert "optional" not in inputs or "local_path" not in inputs.get("optional", {})
 
 
-def test_input_types_model_source_choices():
+def test_input_types_device_choices():
     inputs = OmniVoiceModelLoader.INPUT_TYPES()
-    choices = inputs["required"]["model_source"][0]
-    assert "Auto-download (HuggingFace)" in choices
-    assert "Local path" in choices
+    choices = inputs["required"]["device"][0]
+    assert "cuda:0" in choices
+    assert "cpu" in choices
 
 
 def test_return_type():
     assert OmniVoiceModelLoader.RETURN_TYPES == ("OMNIVOICE_MODEL",)
 
 
-def test_load_model_auto_download():
+def test_load_model():
     loader = OmniVoiceModelLoader()
     mock_model = MagicMock()
     with patch("nodes.loader.OmniVoice") as MockOmniVoice:
         MockOmniVoice.from_pretrained.return_value = mock_model
-        result = loader.load_model(
-            model_source="Auto-download (HuggingFace)",
-            device="cpu",
-            dtype="float32",
-            local_path="",
-        )
+        result = loader.load_model(device="cpu", dtype="float32")
     assert result == (mock_model,)
-    MockOmniVoice.from_pretrained.assert_called_once()
-    call_kwargs = MockOmniVoice.from_pretrained.call_args
-    assert call_kwargs[0][0] == "k2-fsa/OmniVoice"
+    call_args = MockOmniVoice.from_pretrained.call_args
+    assert call_args[0][0] == "k2-fsa/OmniVoice"
 
 
-def test_load_model_local_path():
+def test_load_model_dtype_mapped():
+    import torch
     loader = OmniVoiceModelLoader()
-    mock_model = MagicMock()
     with patch("nodes.loader.OmniVoice") as MockOmniVoice:
-        MockOmniVoice.from_pretrained.return_value = mock_model
-        result = loader.load_model(
-            model_source="Local path",
-            device="cpu",
-            dtype="float32",
-            local_path="/some/local/path",
-        )
-    assert result == (mock_model,)
-    call_args = MockOmniVoice.from_pretrained.call_args[0][0]
-    assert call_args == "/some/local/path"
+        MockOmniVoice.from_pretrained.return_value = MagicMock()
+        loader.load_model(device="cpu", dtype="float16")
+    call_kwargs = MockOmniVoice.from_pretrained.call_args[1]
+    assert call_kwargs["dtype"] == torch.float16