Lemon-cmd/diffusion-jax
Diffusion Probabilistic Model in Jax
This project offers a clear and flexible codebase for experimenting with original Diffusion Probabilistic Models (DDIM and DDPM) to generate images. It takes random noise or existing images and transforms them into new, high-quality images based on a trained model. Image generation researchers and practitioners working on novel diffusion model architectures will find this useful for their experiments.
No commits in the last 6 months.
Use this if you are a machine learning researcher or engineer specifically looking to build, train, and experiment with DDIM and DDPM models for image generation using JAX.
Not ideal if you are looking for an out-of-the-box solution for stable diffusion or general image editing without diving into the model's core mechanics.
Stars
13
Forks
1
Language
Python
License
MIT
Category
Last pushed
Apr 20, 2024
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/diffusion/Lemon-cmd/diffusion-jax"
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.