recruit-jp
/

japanese-clip-vit-b-32-roberta-base

Feature Extraction

Model card Files Files and versions Community

hidehisa-arai commited on Dec 20, 2023

Commit

e7036b0

•

1 Parent(s): f68c08c

update model

Files changed (1) hide show

modeling_japanese_clip.py +1 -0

modeling_japanese_clip.py CHANGED Viewed

@@ -229,6 +229,7 @@ class JapaneseCLIPVisionTransformer(PreTrainedModel):
             act_layer: Callable = nn.GELU,
             norm_layer: Callable = LayerNorm,
             output_tokens: bool = False,
     ):
         super().__init__()
         assert pool_type in ('tok', 'avg', 'none')

             act_layer: Callable = nn.GELU,
             norm_layer: Callable = LayerNorm,
             output_tokens: bool = False,
+            **kwargs,
     ):
         super().__init__()
         assert pool_type in ('tok', 'avg', 'none')