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

Downloads last month
10
Inference Providers NEW
This model is not currently available via any of the supported Inference Providers.
The model cannot be deployed to the HF Inference API: The model has no library tag.