frozenleaves/transformers-mirror
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix CI: Use correct import in SAM for torchvision InterpolationMode (#40160)

Browse Source 
fix ci
This commit is contained in:
Manuel de Prada Corral
2025-08-14 12:53:23 +02:00
committed by GitHub
parent 22e89e5385
commit c47544b16f
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/transformers/models/sam/image_processing_sam_fast.py
View File

@ -243,9 +243,9 @@ class SamImageProcessorFast(BaseImageProcessorFast):
{
"do_normalize": False,
"do_rescale": False,
"interpolation": F.InterpolationMode.NEAREST_EXACT
"interpolation": F_t.InterpolationMode.NEAREST_EXACT
if is_torchvision_v2_available()
else F.InterpolationMode.NEAREST,
else F_t.InterpolationMode.NEAREST,
"size": segmentation_maps_kwargs.pop("mask_size"),
"pad_size": segmentation_maps_kwargs.pop("mask_pad_size"),
}