lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-05 07:29:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update devices.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-31 13:22:40 -08:00
parent fd75f4f101
commit dbd002f930
1 changed files with 1 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
modules/devices.py
View File

@@ -38,15 +38,9 @@ def get_device_for(task):
def torch_gc():
model_management.soft_empty_cache()
if npu_specific.has_npu:
torch_npu_set_device()
npu_specific.torch_npu_gc()
def torch_npu_set_device():
# Work around due to bug in torch_npu, revert me after fixed, @see https://gitee.com/ascend/pytorch/issues/I8KECW?from=project-issue
if npu_specific.has_npu:
torch.npu.set_device(0)
return
def enable_tf32():