mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-10-03 16:45:33 -03:00
Will Miao
35f1ced41a
Add OpenModelDB (openmodeldb.info) as a metadata and download source for
the existing upscaler model type.
Metadata:
- New OpenModelDBClient: fetches the site's bulk JSON dumps, caches them
on disk (24h TTL + ETag revalidation), and builds a local sha256 index
- New OpenModelDBModelMetadataProvider adapts catalogue entries to the
CivitAI-shaped version dict contract; registered in the fallback chain
behind the enable_openmodeldb_api setting (default on), gated to the
upscaler sub-type so other model types never trigger the dump download
- Persisted provenance uses metadata_source "openmodeldb" plus a nested
openmodeldb block (page URL, architecture, scale, license)
Images: paired-image LR/SR URLs are ephemeral imgdiff.net sessions, so
displayable images come from the site-hosted auto-generated thumbnails
(model-level cover leads images[], per-image thumbs for the rest); the
original comparison URL is kept in meta.comparisonUrl.
Downloads:
- New OpenModelDBSource (flat model ids, omdb: group prefix) with
resource filename derivation that recovers names hidden mid-path
(mediafire) or synthesizes {id}.{type} for folder links
- HTML-gateway mirrors (mediafire/mega/drive) are rejected with a clear
manual-download hint instead of silently saving an HTML page as .pth
- ModelSource base gains is_valid_source_id / default_subdir_parts /
resolve_download_url hooks so flat-id sources need no platform branches
UI: "View on OpenModelDB" link in the model modal (downloaded and
hash-enriched models), settings toggle next to the CivArchive one.
58 lines
1.9 KiB
Python
58 lines
1.9 KiB
Python
"""TensorArt model source (link / provenance only).
|
|
|
|
TensorArt support is intentionally limited to *linking* a model to its
|
|
TensorArt page. Automatic metadata extraction is not possible without a
|
|
user session:
|
|
|
|
* ``tensor.art`` sits behind a Cloudflare managed challenge, so plain
|
|
HTTP clients (aiohttp, requests, curl) receive ``403 "Just a moment..."``.
|
|
* Its internal API (``ap-east-1.tensorart.cloud`` / ``cn.tensorart.net``)
|
|
answers every ``/v1/model/*`` route with
|
|
``{"code":100002,"message":"invalid authorization header"}``.
|
|
* The official TAMS API requires an AccessKey/SecretKey pair and request
|
|
signatures, which is a poor fit for a "paste a URL" workflow.
|
|
|
|
``supports_enrichment`` is therefore ``False``: the agent pipeline skips
|
|
these models with an explicit reason instead of failing silently, and the
|
|
UI keeps showing the "View on TensorArt" link. ``tusi.cn`` is TensorArt's
|
|
Chinese mirror and is accepted as the same platform.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import re
|
|
|
|
from .base import ModelSource
|
|
|
|
_DOMAINS = r"(?:tensor\.art|tusi\.cn)"
|
|
|
|
_URL_PATTERN = re.compile(
|
|
rf"https?://(?:www\.)?{_DOMAINS}/models/(?P<id>\d+)"
|
|
)
|
|
|
|
_STRICT_URL_PATTERN = re.compile(
|
|
rf"https?://(?:www\.)?{_DOMAINS}/models/(?P<id>\d+)(?:/[^/?#\s]+)?/?$"
|
|
)
|
|
|
|
|
|
class TensorArtSource(ModelSource):
|
|
"""TensorArt (``tensor.art``)."""
|
|
|
|
platform = "tensorart"
|
|
label = "TensorArt"
|
|
supports_enrichment = False
|
|
supports_download = False
|
|
example_source_id = "827823520299086029"
|
|
url_pattern = _URL_PATTERN
|
|
strict_url_pattern = _STRICT_URL_PATTERN
|
|
|
|
def canonical_url(self, source_id: str) -> str:
|
|
return f"https://tensor.art/models/{source_id}"
|
|
|
|
def asset_base_url(self, source_id: str, revision: str = "") -> str:
|
|
# Unreachable today: enrichment is disabled for this platform.
|
|
return f"https://tensor.art/models/{source_id}"
|
|
|
|
|
|
__all__ = ["TensorArtSource"]
|