This is an astounding job!
I had a little problem with the environment during the reproduction process. I can modify requirements.txt to make it easier for everyone to reproduce
torch==2.2.2
numpy==1.25.2
huggingface_hub==0.26.5
einops==0.8.1
Pillow==11.0.0
torchvision==0.17.2
opencv-python==4.10.0.84
safetensors==0.4.5
matplotlib==3.9.3
# download gpu version of torch-scatter
torch_cluster==1.6.3
-f https://pytorch-geometric.com/whl/torch-2.2.2+cu121.html