Skip to content

Commit 3325b71

Browse files
committed
Fix smoke test
1 parent 2e69d1b commit 3325b71

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/smoke_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@ def smoke_test_torchvision_resnet50_classify(device: str = "cpu") -> None:
5959
model.eval()
6060

6161
# Step 2: Initialize the inference transforms
62-
preprocess = weights.transforms()
62+
preprocess = weights.transforms(antialias=(device != "mps")) # antialias not supported on MPS
6363

6464
# Step 3: Apply inference preprocessing transforms
6565
batch = preprocess(img).unsqueeze(0)

0 commit comments

Comments
 (0)