fix: fix wrong parent class for ColQwen2ForRetrievalOutput

Author: tonywu71

src/transformers/models/colqwen2/modeling_colqwen2.py

@@ -18,7 +18,6 @@
 from typing import List, Optional, Tuple, Union
 
 from ...cache_utils import Cache
-from ...modeling_outputs import BaseModelOutputWithPast
 from ...modeling_utils import PreTrainedModel
 from ...models.auto import AutoModelForImageTextToText
 from ...utils import (
@@ -81,7 +80,7 @@ def _init_weights(self, module):
 
 
 @dataclass
-class ColQwen2ForRetrievalOutput(BaseModelOutputWithPast):
+class ColQwen2ForRetrievalOutput(ModelOutput):
     """
     Base class for ColQwen2 embeddings output.