diff options
Diffstat (limited to 'modules/initialize.py')
| -rw-r--r-- | modules/initialize.py | 5 |
1 files changed, 1 insertions, 4 deletions
diff --git a/modules/initialize.py b/modules/initialize.py index cc34fd6f..f7313ff4 100644 --- a/modules/initialize.py +++ b/modules/initialize.py @@ -143,10 +143,7 @@ def initialize_rest(*, reload_script_modules=False): by that time, so we apply optimization again.
"""
from modules import devices
- # Work around due to bug in torch_npu, revert me after fixed, @see https://gitee.com/ascend/pytorch/issues/I8KECW?from=project-issue
- if devices.npu_specific.has_npu:
- import torch
- torch.npu.set_device(0)
+ devices.torch_npu_set_device()
shared.sd_model # noqa: B018
|