fix-dtype-casting (#13)

- fix-dtype-casting (2011fa2b59111302c5deb3f0791ad31908120430)

modeling_jina_embeddings_v4.py

@@ -345,7 +345,7 @@ class JinaEmbeddingsV4Model(Qwen2_5_VLForConditionalGeneration):
         for batch in tqdm(dataloader, desc=desc):
             with torch.no_grad():
                 batch = {k: v.to(self.device) for k, v in batch.items()}
-                with torch.autocast(device_type=torch.device(self.device).type):
+                with torch.autocast(device_type=torch.device(self.device).type, dtype=torch.bfloat16):
                     embeddings = self(**batch, task_label=task_label)
                     if vector_type == "single_vector":
                         embeddings = embeddings.single_vec_emb