I'll need to check whether passing the downloaded cpsam weights file as the argument to

pretrained_model

will do the right thing.