Instructions to use tianweiy/DMD2 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use tianweiy/DMD2 with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("tianweiy/DMD2", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- Draw Things
- DiffusionBee
Commit History
add 4 step unet safetensors be22767
Tianwei Yin commited on
Update README.md 6743dc8 verified
Update README.md e1cfd4f verified
Update README.md 5e3b0f3 verified
Update README.md f72fab5 verified
fix lora ckpt 19179ac
Tianwei Yin commited on
fix typo 2bd2578
Tianwei Yin commited on
update readme 76cc5d7
Tianwei Yin commited on
update models 552f138
Tianwei Yin commited on
update data 8c752c6
Tianwei Yin commited on
Merge branch 'main' of https://huggingface.co/tianweiy/DMD2 into main a70a18b
Tianwei Yin commited on
update data c37e3bf
Tianwei Yin commited on
Update README.md baf3ffd verified
Update README.md aa7f989 verified
Update README.md 838ac75 verified
add ode pretrained 1-step ckpt c01e8de
Tianwei Yin commited on
update half precision checkpoint 3863dad
Tianwei Yin commited on
update diffuser usage 35dc5c4
Tianwei Yin commited on
move models to main folder b67988e
Tianwei Yin commited on
update link 2c36e07
Tianwei Yin commited on
fix format 476f680
Tianwei Yin commited on
Merge branch 'main' of https://huggingface.co/tianweiy/DMD2 into main 40d8538
Tianwei Yin commited on
update demo link a06ee30
Tianwei Yin commited on
Update README.md 4fb2d86 verified
Update README.md 2e2e265 verified
update readme 1668012
Tianwei Yin commited on
remove demo link for now 3d6cf91
Tianwei Yin commited on
update coco data 5d3045a
Tianwei Yin commited on
update coco test captions 6455775
Tianwei Yin commited on
update laion prompts b58db4c
Tianwei Yin commited on
add sd training datasets f96bab1
Tianwei Yin commited on
update sdv1.5 / sdxl models 4367ce8
Tianwei Yin commited on
upload all imagenet data 48e1bde
Tianwei Yin commited on