Update modeling_qwen2_nomic_vision.py

modeling_qwen2_nomic_vision.py

@@ -773,11 +773,11 @@ QWEN2_INPUTS_DOCSTRING = r"""
 
 @dataclass
 class Qwen2NomicVisionOutput(BaseModelOutputWithPast):
-    last_hidden_state: FloatTensor = None
-    past_key_values: Optional = None
-    hidden_states: Optional = None
-    attentions: Optional = None
-    processed_image: Optional = None
+    last_hidden_state: torch.FloatTensor = None
+    past_key_values: Optional[Tuple[Tuple[torch.FloatTensor]]] = None
+    hidden_states: Optional[Tuple[torch.FloatTensor, ...]] = None
+    attentions: Optional[Tuple[torch.FloatTensor, ...]] = None
+    processed_image: Optional[torch.FloatTensor] = None
 # @add_start_docstrings(
 #     "The bare Qwen2 Model outputting raw hidden-states without any specific head on top.",
 #     QWEN2_START_DOCSTRING,