Adding NPU for get device function (#11617)

* Adding device choice for npu

* Adding device choice for npu

---------

Co-authored-by: J石页 <[email protected]>
Co-authored-by: Sayak Paul <[email protected]>

Author: leisuzz

src/diffusers/utils/torch_utils.py

@@ -18,7 +18,7 @@
 from typing import List, Optional, Tuple, Union
 
 from . import logging
-from .import_utils import is_torch_available, is_torch_version
+from .import_utils import is_torch_available, is_torch_npu_available, is_torch_version
 
 
 if is_torch_available():
@@ -166,6 +166,8 @@ def get_torch_cuda_device_capability():
 def get_device():
     if torch.cuda.is_available():
         return "cuda"
+    elif is_torch_npu_available():
+        return "npu"
     elif hasattr(torch, "xpu") and torch.xpu.is_available():
         return "xpu"
     else: