Update modeling_aimv2.py

modeling_aimv2.py

@@ -222,7 +222,7 @@ class AIMv2Model(AIMv2PretrainedModel):
             hidden_states=hidden_states,
         )
 
-'''
+
 class AIMv2ForImageClassification(AIMv2PretrainedModel):
     def __init__(self, config: AIMv2Config):
         super().__init__(config)
@@ -310,9 +310,8 @@ class AIMv2ForImageClassification(AIMv2PretrainedModel):
             hidden_states=outputs.hidden_states,
             # attentions=outputs.attentions,
         )
-'''
-
 
+'''
 class AIMv2ForImageClassification(AIMv2PretrainedModel):
     def __init__(self, config: AIMv2Config):
         print("Initializing AIMv2ForImageClassification")
@@ -390,5 +389,5 @@ class AIMv2ForImageClassification(AIMv2PretrainedModel):
             loss=loss,
             logits=logits,
             hidden_states=outputs.hidden_states,
-        )
+        )'''