From d7dcfa8917d77c4f7a9d8bc22b7c2a0fb485752d Mon Sep 17 00:00:00 2001
From: Pingzhi Li <pingzhili@mail.ustc.edu.cn>
Date: Mon, 6 Nov 2023 20:16:03 +0800
Subject: [PATCH] Remove an unexpected argument for
 FlaxResNetBasicLayerCollection (#27272)

Remove unexpected argument for FlaxResNetBasicLayerCollection
---
 src/transformers/models/resnet/modeling_flax_resnet.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/src/transformers/models/resnet/modeling_flax_resnet.py b/src/transformers/models/resnet/modeling_flax_resnet.py
index 875716d3f5b..c76da498710 100644
--- a/src/transformers/models/resnet/modeling_flax_resnet.py
+++ b/src/transformers/models/resnet/modeling_flax_resnet.py
@@ -215,7 +215,6 @@ class FlaxResNetBasicLayer(nn.Module):
         self.layer = FlaxResNetBasicLayerCollection(
             out_channels=self.out_channels,
             stride=self.stride,
-            activation=self.activation,
             dtype=self.dtype,
         )
         self.activation_func = ACT2FN[self.activation]