working

modeling_eva_clip.py

@@ -2,6 +2,7 @@ from __future__ import annotations
 
 from collections import OrderedDict
 from copy import deepcopy
+from dataclasses import dataclass
 from typing import Optional, Tuple
 
 import torch
@@ -35,6 +36,7 @@ if not hasattr(nn.Module, "get_parameter_or_buffer"):  # pragma: no cover - comp
     nn.Module.get_parameter_or_buffer = _get_parameter_or_buffer  # type: ignore[attr-defined]
 
 
+@dataclass
 class EvaClipVisionModelOutput(ModelOutput):
     last_hidden_state: torch.FloatTensor