Thank you for your kind response for my previous issue.
I have one more question regarding the different model used for the geometry images and material images.
The model for geometry images is Dit, which flattens the image to 1-D for processing, and the model for material images is imagen, which directly inputs the 2-D images. Why not using the Dit for material images generation?