Skip to content

Commit 5b8dd83

Browse files
authored
Update src/transformers/models/llama4/image_processing_llama4_fast.py
1 parent 3bf26c2 commit 5b8dd83

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

src/transformers/models/llama4/image_processing_llama4_fast.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -414,9 +414,6 @@ def _preprocess(
414414
processed_images = reorder_images(grouped_processed_images, grouped_images_index)
415415
aspect_ratios_list = reorder_images(grouped_aspect_ratios, grouped_images_index)
416416

417-
# temp hack for compatibility test. Not checking in this line
418-
return_tensors = None
419-
420417
processed_images = torch.cat(processed_images, dim=0) if return_tensors else processed_images
421418
aspect_ratios = torch.stack(aspect_ratios_list, dim=0) if return_tensors else aspect_ratios_list
422419
return BatchFeature(

0 commit comments

Comments
 (0)