fix device

Lightning-AI · Jan 9, 2025 · 9690417 · 9690417
1 parent b71fe12
commit 9690417
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/torchmetrics/functional/multimodal/clip_score.py b/src/torchmetrics/functional/multimodal/clip_score.py
@@ -156,7 +156,7 @@ def _clip_score_update(
             f"{len(source_data)} and {len(target_data)}"
         )
 
-    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+    device = torch.device("cpu")
     if source_modality == "image" and isinstance(source_data[0], Tensor):
         device = source_data[0].device
     elif target_modality == "image" and isinstance(target_data[0], Tensor):