diff --git a/modules/intel/ipex/hijacks.py b/modules/intel/ipex/hijacks.py index e2a04a662..a1836b874 100644 --- a/modules/intel/ipex/hijacks.py +++ b/modules/intel/ipex/hijacks.py @@ -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 diff --git a/scripts/postprocessing_pixelart.py b/scripts/postprocessing_pixelart.py index ecd7d41d4..4e6757900 100644 --- a/scripts/postprocessing_pixelart.py +++ b/scripts/postprocessing_pixelart.py @@ -28,14 +28,13 @@ class ScriptPixelArt(scripts_postprocessing.ScriptPostprocessing): if not pixelart_enabled: return from modules.postprocess.pixelart import img_to_pixelart, edge_detect_for_pixelart - device = devices.device if devices.backend != "ipex" else devices.cpu pixel_image = pp.image if pixelart_use_edge_detection: - pixel_image = edge_detect_for_pixelart(pixel_image, image_weight=pixelart_image_weight, block_size=pixelart_edge_block_size, device=device) + pixel_image = edge_detect_for_pixelart(pixel_image, image_weight=pixelart_image_weight, block_size=pixelart_edge_block_size, device=devices.device) pp.info["PixelArt edge block size"] = pixelart_edge_block_size - pixel_image = img_to_pixelart(pixel_image, sharpen=pixelart_sharpen_amount, block_size=pixelart_block_size, device=device) + pixel_image = img_to_pixelart(pixel_image, sharpen=pixelart_sharpen_amount, block_size=pixelart_block_size, device=devices.device) if len(pixel_image) == 1: pixel_image = pixel_image[0] pp.image = pixel_image