jonstreeterandClaude
6dda556567
Repository renamed from ComfyUI-Deep-Exemplar-based-Video-Colorization to ComfyUI-Reference-Based-Video-Colorization to better reflect the comprehensive nature of this toolkit. ## Major Features ### Dual Implementation - ColorMNet (2024) - Modern DINOv2-based approach - Deep Exemplar (2019) - Classic CVPR method - 4 total nodes (2 video, 2 image) ### Modern Components - Multiple feature encoders: VGG19, DINOv2 (ViT-S/B/L), CLIP (ViT-B) - Advanced post-processing: color-matcher, WLS, guided, bilateral filters - Silent auto-installer for dependencies (timm, CLIP, color-matcher) - Comprehensive tooltips on all node parameters ### Performance Optimizations - torch.compile support (15-25% speedup on PyTorch 2.0+) - FP16 mixed precision support - CUDA graphs compatibility handling - Multiple memory modes for ColorMNet ## Bug Fixes - Fix torch.compile CUDA graphs compatibility - Fix ColorMNet model property assignment issue - Remove unsupported disable_cudagraphs parameter - Add proper error handling and graceful fallbacks ## Documentation - Comprehensive user guides and technical documentation - Performance optimization guides - Migration guides - Integration documentation - Auto-installer documentation All URLs updated to: https://github.com/jonstreeter/ComfyUI-Reference-Based-Video-Colorization 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
31 lines
771 B
Plaintext
31 lines
771 B
Plaintext
# ColorMNet ComfyUI Integration Requirements
|
|
# Modern dependencies compatible with ComfyUI
|
|
|
|
# Core dependencies
|
|
torch>=2.0.0
|
|
torchvision>=0.15.0
|
|
numpy>=1.24.0
|
|
Pillow>=9.5.0
|
|
|
|
# Image processing
|
|
scikit-image>=0.21.0
|
|
opencv-python>=4.8.0
|
|
einops>=0.7.0
|
|
|
|
# ColorMNet specific dependencies
|
|
# Note: These need to be installed separately via git
|
|
# py-thin-plate-spline: https://github.com/cheind/py-thin-plate-spline
|
|
# Pytorch-Correlation-extension: https://github.com/ClementPinard/Pytorch-Correlation-extension
|
|
|
|
# Progress bars
|
|
tqdm>=4.65.0
|
|
progressbar2>=4.2.0
|
|
|
|
# Utilities (optional but recommended)
|
|
gdown # For downloading model checkpoints from Google Drive
|
|
|
|
# Development dependencies (optional)
|
|
# pytest>=7.4.0
|
|
# black>=23.0.0
|
|
# mypy>=1.5.0
|