Skip to content

Commit f4da06e

Browse files
committed
fix: Removed use_fast var from img_processor
Signed-off-by: Brandon Groth <[email protected]>
1 parent e7fa61f commit f4da06e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/models/conftest.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1226,7 +1226,7 @@ def process_img(
12261226
torch.FloatTensor: Processed image
12271227
"""
12281228
img_processor = AutoImageProcessor.from_pretrained(pretrained_model)
1229-
batch_dict = img_processor(images=input_img, return_tensor="pt", use_fast=False)
1229+
batch_dict = img_processor(images=input_img, return_tensor="pt")
12301230
# Data is {pixel_values: numpy_array[0]=data} w/ tensor.shape [C,W,H]
12311231
# Needs to be [1,C,W,H] -> unsqueeze(0)
12321232
return torch.from_numpy(batch_dict["pixel_values"][0]).unsqueeze(0)

0 commit comments

Comments
 (0)