Downloads · 30 days
8
17% of all-time downloads
Yuanzhi/DMD2_humble
DMD2_humble is a machine learning model from Yuanzhi. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as apache-2.0.
Downloads · 30 days
8
17% of all-time downloads
All-time downloads
48
Public
Repo size
5.1 GB
Likes
1
Public
Click a slice to open those files.
.safetensors5.1 GB · 100%
From the Hugging Face model README
Home-made DMD2 on laion recap dataset.
rename the safetensor to diffusion_pytorch_model.safetensors
load the unet with unet = UNet2DConditionModel.from_pretrained(current_path).to("cuda", torch.float16)
current_path is the path to the safetensors and config.json
The rest is the same as official DMD2 https://huggingface.co/tianweiy/DMD2