Rename pytorch_model.pth to unimernet_base.pth

#2

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.

Ready to merge
This branch is ready to get merged automatically.

Sign up or log in to comment