Dariusz L
040fa64584
Enhance model linking with CivitAI integration and UI improvements. - __init__.py: Expand model path resolution by building filename→path lookup (including no-ext keys) and using folder_paths to map common categories; add a /model_linker/civitai-search POST route that attempts metadata/hash-based lookup (or filename fallback) and returns CivitAI model/version info for the frontend. - core/sources/civitai.py: Add hashing and metadata handling utilities: _get_sha256_hash, metadata discovery/reading, helpers to extract images/trained-words, get_model_info_by_hash and get_model_info_for_file (which prefers local metadata and otherwise queries CivitAI by hash). Also tidy/normalize existing search/resolve functions and add caching for hash lookups. - web/linker.js: Add a right-click context menu for model chips with actions (Show Info, Open in CivitAI), implement dialog UI to display model info, and wire calls to the new civitai-search endpoint. Includes positioning/closing behavior and dialog population from returned data. These changes enable accurate model identification via local metadata or sha256 lookups on CivitAI and provide convenient frontend actions to view model details or open CivitAI pages.
ComfyUI Model Linker Extension
A ComfyUI extension that helps users relink missing models in workflows using fuzzy matching.
https://github.com/user-attachments/assets/fedf3645-aa66-49f7-b01d-8c3b5127faf4
Features
- Scans all nodes in workflows to find missing models
- Uses fuzzy matching to suggest similar model files
- Updates workflow JSON in UI/memory (user saves themselves)
- Supports all node types
- Optional auto-resolve for 100% confidence matches
Installation
- Clone or download this repository
- Place it in your ComfyUI custom_nodes/ directory
- Restart ComfyUI
Usage
- Open a workflow with missing models
- Click the "🔗 — Model Linker" button in ComfyUI's top menu bar
- Review missing models and their suggested matches
- Click "Resolve" for individual models or "Auto-Resolve 100% Matches" for perfect matches
- Save your workflow when ready
Features
- Subgraph Support: Automatically detects and handles missing models inside subgraphs
- Smart Matching: Shows 100% confidence matches when available, otherwise shows best matches (≥70% confidence)
- Fuzzy Matching: Uses intelligent similarity scoring to find model files even with different naming
- Auto-Resolve: One-click resolution for all perfect matches