smthemex/ComfyUI_FlashVSR

해결하는 문제

이 프로젝트는 고품질 비디오 초해상도를 위해 설계된 고급 확산 기반 프레임워크인 FlashVSR 및 FlashVSR+를 위한 ComfyUI 통합을 제공합니다. 사용자는 비디오를 효율적으로 업스케일할 수 있으며, RTX 4090과 같은 소비자 하드웨어에서 4K 해상도 처리를 지원합니다.

작동 방식

이 프로젝트는 ComfyUI의 사용자 지정 노드 래퍼 역할을 하여 사용자가 FlashVSR 모델을 로드하고 기존 ComfyUI 워크플로우 내에서 실행할 수 있게 합니다. Block-Sparse-Attention 및 Triton 기반 FlashVSR 어텐션을 포함한 여러 어텐션 메커니즘을 지원하여 속도와 메모리 사용량의 균형을 맞춥니다. 사용자는 입력 프레임 수와 VAE 추론 모드와 같은 매개변수를 구성하여 속도 또는 출력 품질을 최적화할 수 있습니다.

대상 사용자

이 도구는 최첨단 확산 모델을 사용하여 고품질 비디오 업스케일링을 수행해야 하는 ComfyUI를 사용하는 비디오 편집자, AI 아티스트 및 연구자를 대상으로 합니다.

주요 기능

• ComfyUI 내에서 실시간 확산 기반 비디오 초해상도를 지원합니다.
• FlashVSR+를 지원하여 효율적이고 고품질의 4K 비디오 업스케일링을 제공합니다.
• VRAM 사용량을 관리하고 메모리 부족(OOM) 오류를 방지하기 위한 유연한 어텐션 구성을 제공합니다.
• 처리 속도와 시각적 충실도 사이의 균형을 위한 다양한 VAE 모드 옵션을 제공합니다.