/OpenDMD-diffusion

Open source implementation and models of One-step Diffusion with Distribution Matching Distillation

Primary LanguagePython

OpenDMD

Hugging Face Models Hugging Face Spaces

Open source implementation and models of One-step Diffusion with Distribution Matching Distillation

image (2) image1 image3 image2

Usage

Launch the gradio demo via:

python gradio_dmd.py

Model Release

These models are very experimental releases, all of them are only trained with very few steps, so the performance is not satisfactory.

Model Link
dreamshaper-8-dmd-1kstep hf-model
dreamshaper-8-dmd-kl-only-6kstep hf-model

Real Time Demo

The video is not accelerated.

demo.mp4

Citation

@article{yin2023one,
  title={One-step Diffusion with Distribution Matching Distillation},
  author={Yin, Tianwei and Gharbi, Micha{\"e}l and Zhang, Richard and Shechtman, Eli and Durand, Fredo and Freeman, William T and Park, Taesung},
  journal={arXiv preprint arXiv:2311.18828},
  year={2023}
}