diff options
| author | Nuullll <vfirst218@gmail.com> | 2024-01-06 19:14:06 +0800 |
|---|---|---|
| committer | Nuullll <vfirst218@gmail.com> | 2024-01-06 19:14:06 +0800 |
| commit | 818d6a11e709bf07d48606bdccab944c46a5f4b0 (patch) | |
| tree | 7b35e955593635a2128237514720432228e58d31 /modules/xpu_specific.py | |
| parent | 73786c047f14d6ae658b2c12f493f05486ba1789 (diff) | |
Fix format
Diffstat (limited to 'modules/xpu_specific.py')
| -rw-r--r-- | modules/xpu_specific.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/modules/xpu_specific.py b/modules/xpu_specific.py index 1137891a..2971dbc3 100644 --- a/modules/xpu_specific.py +++ b/modules/xpu_specific.py @@ -106,8 +106,8 @@ if has_xpu: try: # torch.Generator supports "xpu" device since 2.1 torch.Generator("xpu") - except: - # 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) + except RuntimeError: + # 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', lambda orig_func, device=None: torch.xpu.Generator(device), lambda orig_func, device=None: is_xpu_device(device)) |