Exporting DinoV2

Hello!
Has anyone managed to successfully export the DinoV2 model, specifically the depth estimation head? I’ve been trying for a while now but haven’t had any success. I used the code from the official notebook on GitHub to initialize the model, and for export, I used torch.jit.trace on a preprocessed example image as in the notebook, then the save function, but the trace method crash. I’d like to achieve an export as a .pt file, but ONNX would also be an option.