Set weights_only in torch.load (#36991)

2025-03-27 22:55:50 +08:00
parent de77f5b1ec
commit 41a0e58e5b
28 changed files with 64 additions and 78 deletions
--- a/tests/models/idefics/test_image_processing_idefics.py
+++ b/tests/models/idefics/test_image_processing_idefics.py
@@ -28,7 +28,7 @@ if is_torch_available():
    import torch

 if is_torchvision_available():
-    import torchvision.transforms as transforms
+    from torchvision import transforms

 if is_vision_available():
    from PIL import Image