Error for Fine-Tune SD2 for Inpainting

Keep getting error:
ValueError: Could not complete request to HuggingFace API, Status Code: 500, Error: unknown error, Warnings: ["There was an inference error: __call__() missing 2 required positional arguments: 'image' and 'mask_image'"]
Happens when trying to create new generations for in-painting using a fine-tuned version of Stable Diffusion v2.