TypeError: forward() got an unexpected keyword argument 'early_exit_layers'

[Lsayan]

Hi @zhaoyiran924 , I was going through your work. Thanks for sharing.

I was running the file neuron_detection.py and encountered this error

I investigated a bit and found that the error was originating from utils.py file present in the generation folder in Transformer.

This function is present within the greedy_search function.

Can you please help me resolve this issue? I want to know which function is being invoked here in the above self() call. Thank you very much.

[thangld201]

@Lsayan You need to install the modified transformers package in the repo

[Lsayan]

Thanks for the reply @thangld201 . Actually I was using the modified transformer package present in the repo but still I was receiving this error.
Can you please help me resolve this issue? Thank you for the help