Cannot run inference_on_stl.py script for DoMiNO model

[mrlzchry]

Hi, I had a problem running inference_on_stl.py script on DrivAerML dataset.
This is the error shown,

Then, I turned off the volume sampling from disk option, and the terminal threw this error instead

FYI, this the latest inference_on_stl.py file.

I noticed that in one of its modules that data_mesh is set to None if the world_size =1.
Does it mean that this script cannot work on a single GPU?

It would be the world for me if you can navigate me through this issue.