what's the correct way to load this control net?

#1
by Vinwapgames - opened

im getting OSError: OzzyGT/controlnet-union-promax-sdxl-1.0 does not appear to have a file named diffusion_pytorch_model.bin when doing standard controlnet_model = ControlNetModel.from_pretrained("OzzyGT/controlnet-union-promax-sdxl-1.0" , torch_dtype=torch.float16)

since this is in fp16 and I used the convention to name it, you will need to use the variant arg:

import torch

from diffusers import ControlNetUnionModel


controlnet_model = ControlNetUnionModel.from_pretrained(
    "OzzyGT/controlnet-union-promax-sdxl-1.0",
    torch_dtype=torch.float16,
    variant="fp16",
)

Sign up or log in to comment