Token Classification
GLiNER
PyTorch
multilingual
NER
GLiNER
information extraction
encoder
entity recognition

Error while loading

#1
by bharadwajymg - opened

While loading the code given the code , getting the following issue , kindly help me out
RuntimeError: Error(s) in loading state_dict for SpanModel:
size mismatch for token_rep_layer.bert_layer.model.embeddings.word_embeddings.weight: copying a param with shape torch.Size([128100, 768]) from checkpoint, the shape in current model is torch.Size([128004, 768]).

Knowledgator Engineering org

Please, try to update your gliner package with pip install gliner -U

bharadwajymg changed discussion status to closed

Sign up or log in to comment