How to use other pretrained models for inpainting inference?

[toshi2k2]

How can we use other pretrained models (e.g. lsun_beds256) for inpainting inference? Using the pretrained model trvially seems to be running in the following error -

c = model.cond_stage_model.encode(batch["masked_image"])
AttributeError: 'NoneType' object has no attribute 'encode'