miniDiffusion : Une réimplémentation minimale de Stable Diffusion 3.5 en PyTorch

2025-06-14
miniDiffusion : Une réimplémentation minimale de Stable Diffusion 3.5 en PyTorch

miniDiffusion est une réimplémentation simplifiée du modèle Stable Diffusion 3.5 utilisant PyTorch pur avec un minimum de dépendances. Conçu à des fins éducatives, expérimentales et de hacking, son code concis (~2800 lignes) couvre les scripts VAE, DiT, d'entraînement et de jeu de données. Le projet fournit des scripts pour l'entraînement et l'inférence. Les utilisateurs doivent installer les dépendances et télécharger les poids des modèles pré-entraînés. Ce projet open source est sous licence MIT.

IA