miniDiffusion:PyTorchによるStable Diffusion 3.5の最小限の実装
2025-06-14
miniDiffusionは、最小限の依存関係を持つ純粋なPyTorchを使用してStable Diffusion 3.5モデルを簡素化して再実装したものです。教育、実験、ハッキングの目的で設計されており、簡潔なコードベース(約2800行)には、VAE、DiT、トレーニング、データセットのスクリプトが含まれています。このプロジェクトは、トレーニングと推論の両方のスクリプトを提供します。ユーザーは依存関係をインストールし、事前トレーニングされたモデルの重みをダウンロードする必要があります。このオープンソースプロジェクトはMITライセンスの下でライセンスされています。
AI