bfloat16

illuin-tech · Oct 23, 2024 · 7ead7e7 · 7ead7e7
1 parent ab391b2
commit 7ead7e7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/scripts/configs/flor2/train_colflor_model.yaml b/scripts/configs/flor2/train_colflor_model.yaml
@@ -12,8 +12,8 @@ config:
     (): colpali_engine.utils.transformers_wrappers.AllPurposeWrapper
     class_to_instanciate: !ext colpali_engine.models.ColFlor
     pretrained_model_name_or_path: "./models/ColFlor-base"
-    # torch_dtype:  !ext torch.bfloat16
-    torch_dtype:  !ext torch.float32
+    torch_dtype:  !ext torch.bfloat16
+    # torch_dtype:  !ext torch.float32
     # use_cache: false
 #    device_map: "auto"
 #    quantization_config: