Hi, thanks for your interesting work, I tried to do a inference with your script "sample_image.sh" . It took more than 1H to run the code and finally just got a OOM (I use 1 80G H100). Can you give me more information about the inference time and GPU uasge about running the inference?