How to use a different sampler for inference on a custom model?

[SuroshAhmadZobair]

Hi

I am trying to do inference using my custom-trained model but struggling to implement a different sampler in the inference script.

This is how I am trying to load the model and configs.

from cldm.ddim_hacked import DDIMSampler
model = create_model('./models/cldm_v15.yaml').cpu()
model.load_state_dict(load_state_dict('custom_model.ckpt', location='cuda'))
model = model.cuda()
ddim_sampler = DDIMSampler(model)

I looked at the yaml file but I can't figure out if it should be implemented there.
yaml file: https://huggingface.co/webui/ControlNet-modules-safetensors/blob/main/cldm_v15.yaml

By a different sampling method, I am referring to one of the following:

Any insights?
Thanks in advance.