Agreed. And not just the source, but a fully functional local demo! Runs great on my M1 pro using:
PYTORCH_ENABLE_MPS_FALLBACK=1 python3.10 demo.py --weights checkpoints/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth --device 'mps'
PYTORCH_ENABLE_MPS_FALLBACK=1 python3.10 demo.py --weights checkpoints/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth --device 'mps'
No comments yet.