$ account Comfy-Org
ComfyUI is a modular, node-based AI engine for content creation, primarily focused on generative image, video, audio, and 3D model inference. It provides a web server (aiohttp) with WebSocket and REST APIs, a visual workflow editor, and an extensible plugin system for custom nodes. The architecture is a monolith with a clear separation of concerns: core model loading and sampling (comfy/), execution engine (comfy_execution/), asset management (app/assets/), external API integrations (comfy_api_nodes/), and a public API for defining nodes (comfy_api/). It supports a vast range of open-source diffusion models (Flux, SD3, Wan, Hunyuan, etc.) and offers advanced features like dynamic VRAM management, quantization, multi-GPU support, and experimental guidance methods. The system operates as a local-first application but also integrates with cloud services via partner API nodes. Key entry points include main.py (startup), server.py (PromptServer), and execution.py (PromptExecutor). The codebase demonstrates high craftsmanship in its weight adapter hierarchy (LoRA, LoHa, OFT), patcher extension system for model hooks, and asset management services with database-backed metadata extraction and cursor-based pagination. However, it also contains experimental and deprecated nodes, some code duplication across model implementations, and a reliance on external dependencies that may drift over time.
This repository implements a single ComfyUI custom node, `RTXVideoSuperResolution`, which leverages NVIDIA's RTX Video Super Resolution technology via the `nvidia-vfx` library to upscale images and video frames. The node supports scaling by a multiplier or to target dimensions, and offers quality levels (LOW, MEDIUM, HIGH, ULTRA). It uses batching for efficient GPU memory usage and proper tensor manipulation. The codebase is minimal but well-structured, with a clean schema definition, enum-based resize type selection, and correct integration with the ComfyUI extension API. However, it is tightly coupled to ComfyUI's internal API and the NVIDIA RTX ecosystem, limiting direct reuse outside of that context. No tests, CI/CD beyond publishing, or additional functionality are present.