| 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 |