chandar-lab

/

NeoBERT

like 196

Follow

Chandar Research Lab 67

Hello, I have bug with using Neobert for Sequence Classification- i don't get back logits like in other models. Could you add in line "https://huggingface.co/chandar-lab/NeoBERT/blob/main/model.py#L381"
a code: "return_dict = return_dict if return_dict is not None else self.config.use_return_dict" like it is in other Sequence Classification model, for example microsoft/deberta-v3-base.(They also have this line here: https://github.com/huggingface/transformers/blob/main/src/transformers/models/deberta_v2/modeling_deberta_v2.py#L843)
Thanks