COMET icon indicating copy to clipboard operation
COMET copied to clipboard

Published 20 hours ago • verified publisher icon Unbabel

Add local_files_only parameter to load_from_checkpoint method

Open zzaebok opened this issue 7 months ago • 0 comments

This PR adds local_files_only parameter to load_from_checkpoint method as discussed in https://github.com/Unbabel/COMET/issues/225

The local_files_only parameter is created in the constructor of each encoder and CometModel.

I found that even after downloaded a ckpt file (such as Unbabel/wmt22-comet-da), pretrained_model specified in checkpoint's hparams.yaml should also be downloaded to use tokenizer. So I wrote it in docstring in load_from_checkpoint. Please let me know if there is anything to be fixed :smiley:

zzaebok avatar Jul 29 '24 12:07 zzaebok