#135 · Primary category: MLOps & Evaluation
finetrainers
Scalable and memory-optimized training of diffusion models
Project last updated:05/26/26
GitHub Stars
1.4K
Forks
142
Contributors
24
License
Apache-2.0
Why we included this project
Fine-tuning a modern video or image diffusion model usually means wrestling with VRAM limits, but this library tackles that head-on. Built on the Diffusers stack, it supports supervised fine-tuning and LoRA for models like CogVideoX, LTX-Video, HunyuanVideo, Wan, Flux, and CogView4, and its FP8 weight casting and gradient checkpointing can squeeze some video LoRA runs into roughly 5 GB of VRAM. Training engineers get distributed training across DDP, FSDP-2, and context parallelism, plus a standardized way to define custom model architectures and automatic detection of common dataset layouts. Ready-to-run example scripts and dataset tooling for curating small, high-quality video sets mean you can move from raw clips to a real training run without assembling the plumbing yourself. One caveat: the main branch is actively developed, so pin a stable release tag if you want reproducible results.
Articles for this project
No articles for this project yet.
To suggest a topic or contribute an article, contact us.
Related projects in this category
unsloth
Local UI to run and train LLMs and diffusion models. Supports GGUF, MLX, Qwen3.8, Kimi K3, MiniMax-H3, Gemma 4, FLUX and more.
LlamaFactory
Unified Efficient Fine-Tuning of 100+ LLMs & VLMs (ACL 2024)
airflow
Apache Airflow - A platform to programmatically author, schedule, and monitor workflows
langfuse
🪢 Open source AI engineering platform: LLM evals, observability, metrics, prompt management, playground, datasets. Integrates with OpenTelemetry, LangChain, OpenAI SDK, LiteLLM, and more. 🍊YC W23
netron
Visualizer for neural network, deep learning and machine learning models