Wav2Vec2: Forward() takes 2 positional arguments but 3 were given

I am trying to use wav2vec2_base model for audio feature extraction. I am getting the error in the title for this code:

class Identity(nn.Module):
    @overrides
    def forward(self, input_):
        return input_
            model = wav2vec2_base(num_out=32) 
            model.load_state_dict(torch.load("wav2vec2-base-960h.pt"))
            model.encoder = Identity()
            sample_rate, samples = wavfile.read(wav_file)
            samples = samples.reshape((1,samples.shape[0]))
            samples = torch.from_numpy(samples)
            #samples = samples.type(torch.short)
            print("Start")
            samples = samples.float()

            output = model(samples.float())

I have tried to select the related parts in the code but there is not much left besides these. What is wrong with my forward function? Thanks for any help!

Could you post the entire model definition as well as the error message including the stack trace?