fix: remove useless .to in test

illuin-tech · Sep 10, 2024 · 61ecbd1 · 61ecbd1
1 parent ea3f281
commit 61ecbd1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/models/paligemma/colpali/test_modeling_colpali.py b/tests/models/paligemma/colpali/test_modeling_colpali.py
@@ -84,7 +84,7 @@ def test_colpali_forward_queries(
 
     # Forward pass
     with torch.no_grad():
-        outputs = colpali_from_pretrained(**batch_queries).to(colpali_from_pretrained.device)
+        outputs = colpali_from_pretrained(**batch_queries)
 
     # Assertions
     assert isinstance(outputs, torch.Tensor)