Windecay/ComfyUI-ReservedVRAM

해결하는 문제

이 프로젝트는 워크플로우를 위해 예약된 VRAM의 양을 실시간으로 동적으로 조정할 수 있는 ComfyUI 노드를 제공합니다. 이를 통해 메모리 부족(OOM) 오류를 방지하고 Windows 작업 관리자에서 공유 메모리 오버플로를 관리하여 메모리 제약에 신경 쓰지 않고 GPU 성능을 최대한 활용할 수 있습니다.

작동 방식

이 노드는 ComfyUI의 메모리 관리 시스템과 통합됩니다. EXTRA_RESERVED_VRAM 환경 변수를 업데이트하거나 사용 가능한 경우 model_management.set_extra_reserved_vram() 런타임 API를 사용할 수 있습니다. 또한 DynamicVRAM 및 comfy-aimdo와의 호환성을 지원하여 헤드룸 설정을 동기화합니다. "자동" 모드에서는 노드가 시스템의 "사용 중" VRAM 양을 감지하고 사용자 정의 값을 추가하여 예약량을 계산합니다.

대상 사용자

GPU 메모리를 정밀하게 제어하여 충돌을 방지하고 성능을 최적화해야 하는 ComfyUI 사용자, 특히 여러 프로세스를 실행하거나 고해상도 생성 기능을 사용하는 사용자에게 적합합니다.

주요 기능

• 실시간 조정: 예약된 VRAM 값(GB 단위)을 변경하고 워크플로우 실행 중에 즉시 적용할 수 있습니다.
• 자동 모드: 시스템 VRAM 사용량을 자동으로 감지하여 프로세스가 중단되는 것을 방지합니다.
• VRAM 정리: 출력 전에 VRAM을 지우는 토글이 포함되어 메모리 정리 노드로 작동합니다.
• 무작위 시드 통합: 실행마다 VRAM 전략을 변경하는 무작위 시드 노드로 작동할 수 있습니다.
• 광범위한 호환성: 공식 ComfyUI 빌드 및 DynamicVRAM 및 comfy-aimdo와 같은 특정 확장 기능과 함께 작동합니다.