From dbd002f930fcb97e4836b938afbc5033e9392573 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Wed, 31 Jan 2024 13:22:40 -0800 Subject: [PATCH] Update devices.py --- modules/devices.py | 8 +------- 1 file changed, 1 insertion(+), 7 deletions(-) diff --git a/modules/devices.py b/modules/devices.py index 62dc9f42..7c09d1f4 100644 --- a/modules/devices.py +++ b/modules/devices.py @@ -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():