Not being able to run with pretrained model. This is the error, not sure what checkpoint directory i must provide. I provided both the checkpoint directory and the individual files themselves all cannot work.