概要
このリポジトリは、拡散モデルやフローマッチングモデルといった連続時間生成モデルの開発およびトレーニングコードを集めた「ワークショップ」です。
COCO-2017データセットで1.17億パラメータの拡散モデルを訓練した最新の実験が含まれており、サンプル画像とプロンプトが公開されています。
ユーザーは提供されるJupyter Notebookを通じて、これらのモデルの訓練プロセスやアーキテクチャについて学び、自身の研究や開発に活用できます。
Python、PyTorch、CUDAをベースにしており、画像生成モデルの研究者や開発者が、モデルの動作原理を深く理解し、実践的な実験を行うのに適しています。
プロダクション用途ではなく、あくまで実験的な用途が推奨されます。
互換性・特徴
- Python
- PyTorch
- GPU必須
- Jupyter Notebook
- CLI
- 画像生成
基本情報
| ライセンス | Unlicense |
| Stars | 2 |
| カテゴリ | 画像生成 |
| アクティビティ | low |