入力
| パラメーター | データ型 | Python dtype | 説明 |
|---|---|---|---|
model | MODEL | torch.nn.Module | 連続型 EDM サンプリング機能を付与する対象のモデルです。高度なサンプリング手法を適用するための基盤となります。 |
sampling | COMBO[STRING] | str | 適用するサンプリング方式を指定します。「eps」（イプシロンサンプリング）または「v_prediction」（速度予測）のいずれかを選択でき、サンプリング中のモデル挙動に影響を与えます。 |
sigma_max | FLOAT | float | ノイズレベルの最大シグマ値です。サンプリング時のノイズ注入プロセスにおける上限を制御できます。 |
sigma_min | FLOAT | float | ノイズレベルの最小シグマ値です。ノイズ注入の下限を設定し、モデルのサンプリング精度に影響を与えます。 |
出力
| パラメーター | データ型 | Python dtype | 説明 |
|---|---|---|---|
model | MODEL | torch.nn.Module | 連続型 EDM サンプリング機能を統合した強化済みモデルです。今後の生成タスクでそのまま利用可能です。 |