mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-10-07 10:02:14 -03:00
Will Miao
7b2a108596
Cross-checked both baseModel lists against CivitAI's official baseModelRecords (packages/civitai-shared src/basemodel.constants.ts): - CHECKPOINT_BASE_MODELS gains SD 2.0/2.1 768, SD 2.1 Unclip, SDXL 0.9 / 1.0 LCM / Turbo / Distilled, Playground v2 and Stable Cascade (unCLIP-style but CheckpointLoader-loaded). - DIFFUSION_MODEL_BASE_MODELS gains SVD XT, LTXV 2.5, Flux 3 Video, Wan Image 2.7, Wan Video 2.7 / 3.0, HiDream-O1, Boogu and the Ming Image Design families. API-only (Kling/Sora/Veo/Imagen...), 3D and audio baseModels are intentionally skipped. - Pony V7 exclusion now backed by live-API evidence (model 1901521 is AuraFlow-architecture shipping .gguf variants). CivitAI has no model-level diffusion ModelType: DiT models are uploaded as "Checkpoint" or "UNet", with only uploader-chosen file types to tell them apart. model.type "unet" previously fell through type derivation and failed with 'not supported for download'; it now goes through the checkpoint branch in both the download manager and the download routing endpoint, so the standard chain (file type -> baseModel lists -> unknown default) applies.
121 lines
4.3 KiB
Python
121 lines
4.3 KiB
Python
"""HTTP handler for download target routing decisions."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
import logging
|
|
|
|
from aiohttp import web
|
|
|
|
from ...services.download_routing import (
|
|
is_diffusion_model_download,
|
|
resolve_other_download_sub_type,
|
|
)
|
|
from ...utils.constants import VALID_OTHER_CIVITAI_TYPES
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class DownloadRoutingHandler:
|
|
"""Expose the download-time checkpoint/diffusion-model routing decision.
|
|
|
|
The web UI calls this when the user reaches the download location step
|
|
so the root dropdown offers the same root set (checkpoint vs unet) that
|
|
the download manager would pick for ``use_default_paths``.
|
|
"""
|
|
|
|
async def get_download_routing(self, request: web.Request) -> web.Response:
|
|
try:
|
|
payload = await request.json()
|
|
except json.JSONDecodeError:
|
|
return web.json_response(
|
|
{"success": False, "error": "Invalid JSON payload"}, status=400
|
|
)
|
|
|
|
model_type = payload.get("model_type", "")
|
|
base_model = payload.get("base_model") or ""
|
|
file_types = payload.get("file_types") or []
|
|
selected_file_type = payload.get("selected_file_type")
|
|
|
|
if not isinstance(model_type, str) or not model_type:
|
|
return web.json_response(
|
|
{"success": False, "error": "model_type is required"}, status=400
|
|
)
|
|
if not isinstance(base_model, str) or not isinstance(file_types, list):
|
|
return web.json_response(
|
|
{
|
|
"success": False,
|
|
"error": "base_model must be a string and file_types a list",
|
|
},
|
|
status=400,
|
|
)
|
|
if selected_file_type is not None and not isinstance(selected_file_type, str):
|
|
return web.json_response(
|
|
{"success": False, "error": "selected_file_type must be a string"},
|
|
status=400,
|
|
)
|
|
|
|
# CivitAI ModelType.UNet downloads go through the checkpoint branch,
|
|
# same as in the download manager.
|
|
if model_type.lower() == "unet":
|
|
model_type = "checkpoint"
|
|
|
|
if model_type.lower() in VALID_OTHER_CIVITAI_TYPES:
|
|
from ...services.settings_manager import get_settings_manager
|
|
|
|
settings = get_settings_manager()
|
|
if not settings.is_other_models_enabled():
|
|
# Opt-in feature is off: never auto-route, the UI falls back to
|
|
# manual folder selection and the download manager rejects it.
|
|
return web.json_response(
|
|
{
|
|
"success": True,
|
|
"root_kind": "other",
|
|
"sub_type": None,
|
|
"disabled": True,
|
|
"reason": "other_models_disabled",
|
|
}
|
|
)
|
|
|
|
sub_type = resolve_other_download_sub_type(
|
|
model_type,
|
|
file_types=(str(t) for t in file_types),
|
|
selected_file_type=selected_file_type,
|
|
)
|
|
if sub_type and not settings.is_other_sub_type_enabled(sub_type):
|
|
return web.json_response(
|
|
{
|
|
"success": True,
|
|
"root_kind": "other",
|
|
"sub_type": None,
|
|
"disabled": True,
|
|
"reason": "other_sub_type_disabled",
|
|
"requested_sub_type": sub_type,
|
|
}
|
|
)
|
|
return web.json_response(
|
|
{
|
|
"success": True,
|
|
"root_kind": "other",
|
|
"sub_type": sub_type,
|
|
}
|
|
)
|
|
|
|
from ...services.settings_manager import get_settings_manager
|
|
|
|
is_diffusion = is_diffusion_model_download(
|
|
model_type,
|
|
file_types=(str(t) for t in file_types),
|
|
base_model=base_model,
|
|
unknown_base_model_default=get_settings_manager().get(
|
|
"unknown_base_model_routing", "diffusion_model"
|
|
),
|
|
)
|
|
return web.json_response(
|
|
{
|
|
"success": True,
|
|
"is_diffusion_model": is_diffusion,
|
|
"root_kind": "unet" if is_diffusion else model_type,
|
|
}
|
|
)
|