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>
This commit is contained in:
2026-04-05 18:02:55 +02:00
parent cd0f7aff07
commit b52edcfd84
4 changed files with 21 additions and 48 deletions
+3 -15
View File
@@ -25,10 +25,6 @@ class OmniVoiceModelLoader:
def INPUT_TYPES(cls):
return {
"required": {
"model_source": (
["Auto-download (HuggingFace)", "Local path"],
{"default": "Auto-download (HuggingFace)"},
),
"device": (
["cuda:0", "cuda:1", "cpu"],
{"default": "cuda:0"},
@@ -38,9 +34,6 @@ class OmniVoiceModelLoader:
{"default": "float16"},
),
},
"optional": {
"local_path": ("STRING", {"default": ""}),
},
}
RETURN_TYPES = ("OMNIVOICE_MODEL",)
@@ -48,19 +41,14 @@ class OmniVoiceModelLoader:
FUNCTION = "load_model"
CATEGORY = "OmniVoice"
def load_model(self, model_source, device, dtype, local_path=""):
def load_model(self, device, dtype):
if OmniVoice is None:
raise ImportError(
"omnivoice is not installed. Run: pip install omnivoice"
"omnivoice is not installed. Run: pip install omnivoice --no-deps"
)
if model_source == "Local path" and local_path:
source = local_path
else:
source = "k2-fsa/OmniVoice"
model = OmniVoice.from_pretrained(
source,
"k2-fsa/OmniVoice",
device_map=device,
dtype=DTYPE_MAP[dtype],
cache_dir=CACHE_DIR,