Rename pytorch_model.pth to unimernet_base.pth
#2
by
MustaphaUnubiMomoh
- opened
The configuration file i.e., demo.yaml in the UniMERNet repository on github expects model file ./models/unimernet_base/unimernet_base.pth but the actual path is ./models/unimernet_base/pytorch_model.pth.
This leads to a RuntimeError:
RuntimeError("checkpoint url or path is invalid")
Solution: rename pytorch_model.pth to unimernet_base.pth to ensure the model loading process can successfully locate the checkpoint file as specified in the configuration.