feat: fix the default values

modeling_jina_embeddings_v4.py

@@ -288,8 +288,9 @@ class JinaEmbeddingsV4Model(Qwen2_5_VLForConditionalGeneration):
             attention_mask (torch.Tensor): The attention mask tensor.
         Returns:
             JinaEmbeddingsV4ModelOutput:
-                single (torch.Tensor): Single-vector embeddings of shape (batch_size, dim).
-                multi (torch.Tensor): Multi-vector embeddings of shape (batch_size, num_tokens, dim).
+                vlm_last_hidden_states (torch.Tensor, optional): Last hidden states of the VLM.
+                single_vec_emb (torch.Tensor, optional): Single-vector embeddings.
+                multi_vec_emb (torch.Tensor, optional): Multi-vector embeddings.
         """
         # Forward pass through the VLM
         hidden_states = self.get_last_hidden_states(
@@ -362,7 +363,7 @@ class JinaEmbeddingsV4Model(Qwen2_5_VLForConditionalGeneration):
 
     def _validate_encoding_params(
         self,
-        output_format: Optional[Union[str, VectorOutputFormat]] = VectorOutputFormat.SINGLE,
+        output_format: Optional[Union[str, VectorOutputFormat]] = None,
         truncate_dim: Optional[int] = None,
         prompt_name: Optional[str] = None,
     ) -> Dict[str, Any]: