Diffusion Model Loader KJ (DiffusionModelLoaderKJ) - ComfyUI-KJNodes

Diffusion Model Loader KJ ノードは、モデル読み込み時に、重みデータ型と計算データ型に対する詳細な制御を提供し、fp8からfp32までの様々な精度フォーマットをサポートします。

Diffusion Model Loader KJは、拡散モデル（Stable Diffusionなど）を読み込み、設定するためのノードです。その中核機能は、指定されたパスからモデルファイルを読み込み、実行時の数値精度や計算動作をユーザーが調整して、パフォーマンスや互換性を最適化できるようにすることです。

ノード機能

このノードは、モデル読み込み時に、重みデータ型と計算データ型に対する詳細な制御を提供し、fp8からfp32までの様々な精度フォーマットをサポートします。主な技術的特徴には、特定のGPUアクセラレーションライブラリを活用するためのオプションのCublasLinearパッチ、SageAttentionメカニズムの統合、追加の状態辞書を通じてモデル機能を統合または強化するサポート（例：ビデオモデルにVACEモジュールを追加）などが含まれます。

ノードパラメータ説明 - Diffusion Model Loader KJ

接続入力 (Inputs)

パラメータ名	データ型	必須	デフォルト値	範囲/オプション	説明
`model_name`	COMBO	はい	-	-	読み込む拡散モデルファイルを選択します。このパラメータは、ComfyUIが管理するモデルディレクトリ内の利用可能なすべてのモデルを一覧表示します。

コントロールパラメータ (Parameters)

パラメータ名	データ型	必須	デフォルト値	範囲/オプション	説明
`weight_dtype`	COMBO	はい	default	default, fp8_e4m3fn, fp8_e4m3fn_fast, fp8_e5m2, fp16, bf16, fp32	メモリ使用量と計算精度のバランスを取るために使用します。
`compute_dtype`	COMBO	はい	default	default, fp16, bf16, fp32	計算速度と数値安定性に影響します。
`patch_cublaslinear`	BOOLEAN	はい	False	-	cublas_ops引数の有効化または無効化
`sage_attention`	COMBO	はい	False	false, true	ComfyUIのアテンションをパッチしてsageattnを使用します。
`enable_fp16_accumulation`	BOOLEAN	はい	False	-	torch.backends.cuda.matmul.allow_fp16_accumulationを有効にします。PyTorch 2.7.0 ナイトリーが必要です。
`extra_state_dict`	STRING	いいえ	-	-	読み込む追加の状態辞書へのフルパス。これはメインの状態辞書とマージされます。例えば、WanVideoModelにVACEモジュールを追加する場合などに便利です。DiffusionModelSelectorを使用して簡単にパスを取得できます。

出力 (Output)

パラメータ名	データ型	説明
MODEL	MODEL	読み込まれ設定されたモデルを出力し、後続のノードで使用できるようにします。

使用シナリオ

ベースモデルを読み込み、特定の最適化や拡張機能を適用する必要があるワークフローで、このノードは非常に有用です。例えば、このノードを使用してWanVideoModelを読み込み、extra_state_dictパラメータを通じてVACEモジュールの状態辞書をマージすることで、機能が強化されたビデオ生成モデルを構築できます。

注意事項

このノードは実験的カテゴリに属し、一部の機能（enable_fp16_accumulationなど）は、特定のバージョンのPyTorch（2.7.0 ナイトリーなど）で正常に動作する必要があります。高度なデータ型（fp8など）やパッチ機能を使用する前に、ハードウェアとソフトウェア環境がサポートしていることを確認してください。

Diffusion Model Loader KJ ノードソースコードリンク

Diffusion Model Loader KJ ノードは ComfyUI-KJNodes ノードパッケージから提供されています。

Diffusion Model Loader KJ(DiffusionModelLoaderKJ)