TypeError: model_fn_joyai_image() missing 3 required positional arguments: 'latents', 'prompt_embeds', and 'prompt_embeds_mask'

[dongdk]

when run the examples/joyai_image/model_training/lora/JoyAI-Image-Edit.sh, it encounters this problems, but run examples/joyai_image/model_inference/JoyAI-Image-Edit.py successfully

[mi804]

It might be that the dataset was not downloaded correctly. Please refer to the script to download the sample dataset. Additionally, this is a two-stage training example, so make sure that cache data has been generated in models/train/JoyAI-Image-Edit-split-cache.