cabralpinto/modular-diffusion
Python library for designing and training your own Diffusion Models with PyTorch
This tool helps machine learning researchers and enthusiasts create custom Diffusion Models for generating new data, like high-quality images. You provide your existing dataset and define the model's components, and it outputs a trained model capable of generating new, similar data. It's designed for those who want to experiment with or build novel generative AI solutions.
291 stars. No commits in the last 6 months. Available on PyPI.
Use this if you are a machine learning researcher or enthusiast who wants to design, train, and experiment with your own Diffusion Models using PyTorch.
Not ideal if you are looking for a pre-trained, ready-to-use Diffusion Model or if you are not comfortable with Python and PyTorch.
Stars
291
Forks
14
Language
Python
License
MIT
Category
Last pushed
Jun 17, 2025
Commits (30d)
0
Dependencies
3
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/diffusion/cabralpinto/modular-diffusion"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Higher-rated alternatives
huggingface/diffusers
🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch.
bghira/SimpleTuner
A general fine-tuning kit geared toward image/video/audio diffusion models.
mcmonkeyprojects/SwarmUI
SwarmUI (formerly StableSwarmUI), A Modular Stable Diffusion Web-User-Interface, with an...
nateraw/stable-diffusion-videos
Create 🔥 videos with Stable Diffusion by exploring the latent space and morphing between text prompts
TheDesignFounder/DreamLayer
Benchmark diffusion models faster. Automate evals, seeds, and metrics for reproducible results.