From af8121ec22de4c218ab470500c0c0e2f2127eb38 Mon Sep 17 00:00:00 2001 From: Clint Adams Date: Sun, 7 Sep 2025 11:30:28 -0400 Subject: [PATCH] Set accepts_loss_kwargs to False for ConvNext(|V2)ForImageClassification --- src/transformers/models/convnext/modeling_convnext.py | 2 ++ src/transformers/models/convnextv2/modeling_convnextv2.py | 2 ++ 2 files changed, 4 insertions(+) diff --git a/src/transformers/models/convnext/modeling_convnext.py b/src/transformers/models/convnext/modeling_convnext.py index f89134f803f2..b61d18ed7294 100755 --- a/src/transformers/models/convnext/modeling_convnext.py +++ b/src/transformers/models/convnext/modeling_convnext.py @@ -310,6 +310,8 @@ def forward( """ ) class ConvNextForImageClassification(ConvNextPreTrainedModel): + accepts_loss_kwargs = False + def __init__(self, config): super().__init__(config) diff --git a/src/transformers/models/convnextv2/modeling_convnextv2.py b/src/transformers/models/convnextv2/modeling_convnextv2.py index 7c774eb26225..a5fe43f84c8b 100644 --- a/src/transformers/models/convnextv2/modeling_convnextv2.py +++ b/src/transformers/models/convnextv2/modeling_convnextv2.py @@ -332,6 +332,8 @@ def forward( ) # Copied from transformers.models.convnext.modeling_convnext.ConvNextForImageClassification with CONVNEXT->CONVNEXTV2,ConvNext->ConvNextV2,convnext->convnextv2 class ConvNextV2ForImageClassification(ConvNextV2PreTrainedModel): + accepts_loss_kwargs = False + def __init__(self, config): super().__init__(config)