mirror of
https://github.com/kohya-ss/sd-scripts.git
synced 2026-04-09 06:45:09 +00:00
IPEX fix torch.UntypedStorage.is_cuda
This commit is contained in:
@@ -117,6 +117,7 @@ def linalg_solve(A, B, *args, **kwargs): # pylint: disable=invalid-name
|
|||||||
else:
|
else:
|
||||||
return original_linalg_solve(A, B, *args, **kwargs)
|
return original_linalg_solve(A, B, *args, **kwargs)
|
||||||
|
|
||||||
|
@property
|
||||||
def is_cuda(self):
|
def is_cuda(self):
|
||||||
return self.device.type == 'xpu'
|
return self.device.type == 'xpu'
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user