From fa2917a6bec41487aefc5436c3cb7696fb4159ab Mon Sep 17 00:00:00 2001 From: Disty0 Date: Sat, 10 Jan 2026 20:48:36 +0300 Subject: [PATCH] cleanup --- modules/intel/ipex/hijacks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/intel/ipex/hijacks.py b/modules/intel/ipex/hijacks.py index 5537ba64e..327d9f18a 100644 --- a/modules/intel/ipex/hijacks.py +++ b/modules/intel/ipex/hijacks.py @@ -70,7 +70,7 @@ original_get_autocast_dtype = torch.get_autocast_dtype @wraps(torch.get_autocast_dtype) def torch_get_autocast_dtype(device_type=None): if device_type is None or check_cuda(device_type) or check_device_type(device_type, "xpu"): - return torch.bfloat16 + return devices.dtype or torch.bfloat16 else: return original_get_autocast_dtype(device_type)