diff options
| author | wangshuai09 <391746016@qq.com> | 2024-01-31 10:46:53 +0800 |
|---|---|---|
| committer | wangshuai09 <391746016@qq.com> | 2024-01-31 12:29:58 +0800 |
| commit | cc3f604310458eed7d26456c1b3934d582283ffe (patch) | |
| tree | 44cbaa680fd3afeea5799fa75af4d52872f646ba /modules/initialize.py | |
| parent | 74ff85a1a1ee4cce432b1c7d33c1eda831f68d48 (diff) | |
Update
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
|