fast-stable-diffusion  by TheLastBen

Colab notebooks for fast Stable Diffusion + DreamBooth

Created 3 years ago
7,840 stars

Top 6.7% on SourcePulse

GitHubView on GitHub
Project Summary

This repository provides optimized implementations for Stable Diffusion and DreamBooth training, targeting users seeking faster and more efficient image generation and fine-tuning. It offers Colab notebooks for quick experimentation and aims to streamline the process for researchers and hobbyists.

How It Works

The project leverages optimized CUDA kernels and efficient training loops to accelerate Stable Diffusion and DreamBooth workflows. It focuses on reducing VRAM usage and increasing training speed, enabling users to fine-tune models on consumer-grade hardware more effectively.

Quick Start & Requirements

Highlighted Details

  • Optimized CUDA kernels for faster training.
  • Reduced VRAM requirements for DreamBooth.
  • Streamlined Colab notebooks for ease of use.

Maintenance & Community

  • The author mentions offering paid, advanced trainers and Colab notebooks via X (Twitter).
  • Maintenance for Paperspace notebooks will cease.

Licensing & Compatibility

  • The repository's license is not explicitly stated in the provided README snippet.

Limitations & Caveats

  • The project's primary focus is on Colab notebooks, with potential limited support for other environments.
  • Maintenance for specific platforms (Paperspace) is being discontinued.
Health Check
Last Commit

3 weeks ago

Responsiveness

1 day

Pull Requests (30d)
0
Issues (30d)
17
Star History
39 stars in the last 30 days

Explore Similar Projects

Starred by Chip Huyen Chip Huyen(Author of "AI Engineering", "Designing Machine Learning Systems") and Ying Sheng Ying Sheng(Coauthor of SGLang).

fastllm by ztxz16

0.4%
4k
High-performance C++ LLM inference library
Created 2 years ago
Updated 1 week ago
Starred by François Chollet François Chollet(Author of Keras; Cofounder of Ndea, ARC Prize), Chaoyu Yang Chaoyu Yang(Founder of Bento), and
13 more.

neon by NervanaSystems

0%
4k
Deep learning framework (discontinued)
Created 11 years ago
Updated 4 years ago
Starred by Clement Delangue Clement Delangue(Cofounder of Hugging Face), Chip Huyen Chip Huyen(Author of "AI Engineering", "Designing Machine Learning Systems"), and
20 more.

accelerate by huggingface

0.3%
9k
PyTorch training helper for distributed execution
Created 4 years ago
Updated 1 day ago
Starred by Andrej Karpathy Andrej Karpathy(Founder of Eureka Labs; Formerly at Tesla, OpenAI; Author of CS 231n), Jiayi Pan Jiayi Pan(Author of SWE-Gym; MTS at xAI), and
34 more.

flash-attention by Dao-AILab

0.6%
20k
Fast, memory-efficient attention implementation
Created 3 years ago
Updated 1 day ago
Feedback? Help us improve.