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などの特定の拡張機能と連携します。