File size: 370 Bytes
d2f23e3 b2609c2 d2f23e3 b2609c2 d2f23e3 b2609c2 d2f23e3 b2609c2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
---
license: apache-2.0
---
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 |