test does not use cuda

dnth · Nov 9, 2024 · 6d03fc0 · 6d03fc0
1 parent aa41b34
commit 6d03fc0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/smoke/test_timm.py b/tests/smoke/test_timm.py
@@ -9,7 +9,7 @@
 
 @pytest.fixture
 def model():
-    return xinfer.create_model("timm/resnet18.a1_in1k", device="cuda", dtype="float16")
+    return xinfer.create_model("timm/resnet18.a1_in1k")
 
 
 @pytest.fixture