issue with ConvNext
#1
by
gagannarula
- opened
Hi i tried to use the AudioProtoPNet model following the code snippet in the README, and got this error:
TypeError: ConvNextModel.forward() takes from 1 to 3 positional arguments but 4 were given
It looks like return_dict shouldn't be passed to the self.backbone which is ConvNextModel. The issue maybe that I am using the latest version of transformers. I solved the problem locally by removing return_dict argument on line 888:
return self.backbone(input_values, output_hidden_states) # , return_dict)