Skip to content

S
Stable Diffusion Webui
Commit 818d6a11 authored by Nuullll's avatar Nuullll
Browse files
Options

Fix format

parent 73786c04
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
modules/xpu_specific.py
View file @ 818d6a11
...@@ -106,8 +106,8 @@ if has_xpu: ...@@ -106,8 +106,8 @@ if has_xpu:
try: try:
# torch.Generator supports "xpu" device since 2.1 # torch.Generator supports "xpu" device since 2.1
torch.Generator("xpu") torch.Generator("xpu")
except: except RuntimeError:
# W/A for https://github.com/intel/intel-extension-for-pytorch/issues/452: torch.Generator API doesn't support XPU device (for IPEX < 2.1) # W/A for https://github.com/intel/intel-extension-for-pytorch/issues/452: torch.Generator API doesn't support XPU device (for torch < 2.1)
CondFunc('torch.Generator', CondFunc('torch.Generator',
lambda orig_func, device=None: torch.xpu.Generator(device), lambda orig_func, device=None: torch.xpu.Generator(device),
lambda orig_func, device=None: is_xpu_device(device)) lambda orig_func, device=None: is_xpu_device(device))
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment