Celestium Neonは、GTX 1650のようにメモリがわずか4GBのグラフィックスカードでStable Diffusion XL (SDXL)推論を可能にし、VRAMをクリーンアップするために設計された新しいツールです。

このツールは以下の主要なメカニズムを通じて動作します:

  • ゾンビテンソルからVRAMを解放するためのL1-L2 PyTorchクリーンアップ
  • MiBの低下をリアルタイムで表示するリアルタイムVRAMモニタリング
  • メモリ不足エラーが発生したときにクリーンアップを強制するSDXL RESET機能
  • 5分ごとに設定される自動クリーンアップ間隔

ドライバー591.86を搭載したGTX 1650でのテストにより、このツールがOOMクラッシュを成功裡に防止し、クリーンアップ後にVRAM使用量を938 MiBまで削減して画像生成の完了を可能にすることが示されました。