概要
LeapAlignは、フローマッチングモデルを人間の好みに合わせて微調整するための学習後手法です。
完全なサンプリング軌道に勾配を逆伝播させる際のメモリ消費と勾配爆発の問題を解決するため、2段階の「リープ軌道」を構築し、任意の生成ステップに勾配を効率的に伝播させます。
これにより、特に画像生成の初期段階で全体の構造を決定するステップの最適化が容易になります。
画像生成モデルの品質を向上させたい研究者や開発者、特にPyTorchとGPU環境を利用するユーザーを想定しています。
互換性・特徴
- PyTorch
- Python
- GPU必須
- CLI
- Hugging Face対応
- 画像生成
基本情報
| ライセンス | Apache-2.0 |
| Stars | 25 |
| Forks | 1 |
| カテゴリ | 画像生成 |
| アクティビティ | low |