diff --git a/extensions_built_in/llm_models/qwen25_omni.py b/extensions_built_in/llm_models/qwen25_omni.py index 87374b1..cbf29a1 100644 --- a/extensions_built_in/llm_models/qwen25_omni.py +++ b/extensions_built_in/llm_models/qwen25_omni.py @@ -45,7 +45,7 @@ BASE_REPO = "Qwen/Qwen2.5-Omni-7B" # thinker hidden size -> config/processor repo; single-file checkpoints carry no config BASE_REPO_BY_HIDDEN = {3584: "Qwen/Qwen2.5-Omni-7B", 2048: "Qwen/Qwen2.5-Omni-3B"} # single-file convrot8 thinker written by scripts/convert_vllm_to_comfy.py -DEFAULT_CHECKPOINT = "ostris/qwen2_5_omni/text_encoders/qwen2_5_omni_7b_convrot8.safetensors" +DEFAULT_CHECKPOINT = "ostris/Qwen2.5-Omni-7B/qwen2_5_omni_7b_convrot8.safetensors" DEFAULT_INSTRUCTION = "Describe this in detail." SAMPLE_RATE = 16000 diff --git a/toolkit/models/registry.py b/toolkit/models/registry.py index 403aca7..0133426 100644 --- a/toolkit/models/registry.py +++ b/toolkit/models/registry.py @@ -55,7 +55,7 @@ ARCH_REGISTRY: Dict[str, dict] = { "ltx2.3": {"modality": "video", "model": {"name_or_path": "Lightricks/LTX-2.3/ltx-2.3-22b-dev.safetensors", "quantize": True, "quantize_te": True}, "sample": {**VID, "num_inference_steps": 25, "guidance_scale": 3.0, "num_frames": 49, "fps": 24}}, "ltx2.5": {"modality": "video", "model": {"name_or_path": "Lightricks/LTX-2.5", "quantize": True, "quantize_te": True}, "sample": {**VID, "num_inference_steps": 25, "guidance_scale": 3.0, "num_frames": 49, "fps": 24}}, "ace_step_15": {"modality": "audio", "model": {"name_or_path": "ostris/ace_step_1.5_ComfyUI_files/ace_step_1.5_base_aio.safetensors", "quantize": True, "quantize_te": True}, "sample": {"width": 512, "height": 512, "num_inference_steps": 20, "guidance_scale": 4.0}}, - "qwen25_omni": {"modality": "text", "model": {"name_or_path": "ostris/qwen2_5_omni/text_encoders/qwen2_5_omni_7b_convrot8.safetensors", "quantize": True, "qtype": "convrot8"}, "sample": {"width": 512, "height": 512, "num_inference_steps": 1, "guidance_scale": 1.0}}, + "qwen25_omni": {"modality": "text", "model": {"name_or_path": "ostris/Qwen2.5-Omni-7B/qwen2_5_omni_7b_convrot8.safetensors", "quantize": True, "qtype": "convrot8"}, "sample": {"width": 512, "height": 512, "num_inference_steps": 1, "guidance_scale": 1.0}}, "yue2": {"modality": "audio", "model": {"name_or_path": "Comfy-Org/YuE2/checkpoints/yue2_3b_int8_convrot.safetensors", "quantize": True, "qtype": "convrot8"}, "sample": {"width": 512, "height": 512, "num_inference_steps": 32, "guidance_scale": 1.0, "duration": 120}}, } diff --git a/ui/src/app/jobs/new/options.tsx b/ui/src/app/jobs/new/options.tsx index 4f53e1c..78844fb 100644 --- a/ui/src/app/jobs/new/options.tsx +++ b/ui/src/app/jobs/new/options.tsx @@ -1387,7 +1387,7 @@ export const modelArchs: ModelArch[] = [ defaults: { // default updates when [selected, unselected] in the UI 'config.process[0].model.name_or_path': [ - 'ostris/qwen2_5_omni/text_encoders/qwen2_5_omni_7b_convrot8.safetensors', + 'ostris/Qwen2.5-Omni-7B/qwen2_5_omni_7b_convrot8.safetensors', defaultNameOrPath, ], 'config.process[0].model.quantize': [true, false],