Skip to content

Conversation

@WangYangfan
Copy link

load local models for bert-score.

just change model_type to your local model path, e.g.

self.bert_scorer = BERTScorer(
    model_type="/Users/wangyangfan/remote/models/FacebookAI/roberta-large",
    device='mps', 
    rescale_with_baseline=True, 
    lang='en'
)

👏 Inspired by @shuaijiumei, thanks a lot!

load local models for bert-score
@WangYangfan WangYangfan changed the title Update scorer.py load local models for bert-score. Jun 14, 2024
@chuyuanli
Copy link

Seems also need to specify num_layers to make it work properly

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants