search

barthelemymp / TULIP-TCR

GNU General Public License v3.0
8 stars 3 forks source link

TypeError: ED_LMOutput.__init__() got an unexpected keyword argument 'loss' #11

Open SteamedGit opened 2 months ago

SteamedGit commented 2 months ago

This error occurs after adding the dataclass decorator to ED_LMOutput. I suspect that 'loss' should be 'clf_loss' but I'm not sure since I'm unfamiliar with the repo.

I added in the dataclass decorator since without it you get the error: "TypeError: src.multiTrans.ED_LMOutput is not a dataclasss. This is a subclass of ModelOutput and so must use the @dataclass decorator."

I suspect that there have been changes made to Huggingface's transformers that break this repo.

barthelemymp commented 2 months ago

Hi,

Please use pip install transformers==4.32.1

Best

Barthelemy

SteamedGit commented 2 months ago

Thanks! Could you please provide version numbers for the other dependencies as well?