Ben FinkleaandClaude Opus 4.5
9256a1c87c
A custom node extension for ComfyUI that provides: - Model version tracking for HuggingFace and CivitAI - Automatic model updates with configurable scheduling - Multi-GPU synchronization with VRAM-based filtering - Metadata management with hash verification Features: - SQLite database for model metadata storage - REST API endpoints for all operations - Web UI panel integrated into ComfyUI menu - Background scheduler for automatic update checks - rsync-based sync to remote machines - VRAM estimation for model filtering Tests: - 17 unit tests covering all library modules - Playwright integration tests for UI verification Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
36 lines
915 B
Python
36 lines
915 B
Python
"""
|
|
ComfyUI-ModelSync: Model Auto-Update & Multi-GPU Sync
|
|
|
|
A custom node extension for ComfyUI that provides:
|
|
- Model version tracking for HuggingFace and CivitAI
|
|
- Automatic model updates with configurable scheduling
|
|
- Multi-GPU synchronization with VRAM-based filtering
|
|
- Metadata management with hash verification
|
|
|
|
Repository: https://github.com/benfinklea/ComfyUI-ModelSync
|
|
License: MIT
|
|
"""
|
|
|
|
__version__ = "0.1.0"
|
|
__author__ = "Ben Finklea"
|
|
|
|
# Node class mappings (empty - this extension provides API endpoints, not nodes)
|
|
NODE_CLASS_MAPPINGS = {}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {}
|
|
|
|
# Web extension directory
|
|
WEB_DIRECTORY = "./js"
|
|
|
|
|
|
def setup_api():
|
|
"""Initialize the ModelSync API endpoints."""
|
|
try:
|
|
from .server import setup_routes
|
|
setup_routes()
|
|
except Exception as e:
|
|
print(f"[ModelSync] Failed to setup API routes: {e}")
|
|
|
|
|
|
# Initialize when loaded by ComfyUI
|
|
setup_api()
|