From 5bacb048dd03d82a671d3731843fe6e23c70b4b0 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Thu, 6 Mar 2025 13:10:20 +0100 Subject: [PATCH] More adjustments --- embedding_converter/src/training.py | 1 - 1 file changed, 1 deletion(-) diff --git a/embedding_converter/src/training.py b/embedding_converter/src/training.py index 4bf5b78..6012936 100644 --- a/embedding_converter/src/training.py +++ b/embedding_converter/src/training.py @@ -139,7 +139,6 @@ def train() -> None: } } - if torch.cuda.is_available(): torch.set_float32_matmul_precision('high')