IPEX fix bicubic upsample with torchvision

This commit is contained in:
Disty0
2025-07-06 18:19:55 +03:00
parent 526976eb64
commit 148b011692
2 changed files with 8 additions and 3 deletions
+6
View File
@@ -452,6 +452,12 @@ def ipex_hijacks():
torch.from_numpy = from_numpy
torch.as_tensor = as_tensor
try:
import torchvision
torchvision.transforms._functional_tensor.interpolate = interpolate
except Exception:
pass
# AMP:
torch.amp.grad_scaler.GradScaler.__init__ = GradScaler_init
torch.is_autocast_enabled = torch_is_autocast_enabled