Update README.md

README.md

@@ -71,9 +71,9 @@ import torch
 
 
 MODEL_PATH = "vidore/colpali-v1.3-hf"
-VARIANT = "vision"                     # one of "vision" or "text"
+VARIANT = "text"                        # one of "vision" or "text"
 ONNX_PATH = f"onnx/{VARIANT}/model.onnx"
-MODEL_DTYPE = torch.float32             # one of torch.float32 or torch.float16
+MODEL_DTYPE = torch.float16             # one of torch.float32 or torch.float16
 
 base_model = ColPaliForRetrieval.from_pretrained(MODEL_PATH)
 base_model = base_model.to(dtype=MODEL_DTYPE)