We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e7fa61f commit f4da06eCopy full SHA for f4da06e
tests/models/conftest.py
@@ -1226,7 +1226,7 @@ def process_img(
1226
torch.FloatTensor: Processed image
1227
"""
1228
img_processor = AutoImageProcessor.from_pretrained(pretrained_model)
1229
- batch_dict = img_processor(images=input_img, return_tensor="pt", use_fast=False)
+ batch_dict = img_processor(images=input_img, return_tensor="pt")
1230
# Data is {pixel_values: numpy_array[0]=data} w/ tensor.shape [C,W,H]
1231
# Needs to be [1,C,W,H] -> unsqueeze(0)
1232
return torch.from_numpy(batch_dict["pixel_values"][0]).unsqueeze(0)
0 commit comments