Directory structure mismatch in JoyAI-Image-Und (missing text_encoder/)

I found a directory structure issue when running inference_und.py.
The script expects a text_encoder/ subdirectory inside the --ckpt-root, but the files downloaded from Hugging Face (JoyAI-Image-Und) are stored in a flat structure at the root level.
Because of this, it throws:
FileNotFoundError: Expected text_encoder/ directory inside checkpoint root
Could you please update the HF repo structure or the inference script? Also, I couldn't find the main model.pt in the JoyAI-Image-Und folder.