From b4c10ce37d8799930d53de1475b42582f0857a6b Mon Sep 17 00:00:00 2001
From: RVC-Boss <129054828+RVC-Boss@users.noreply.github.com>
Date: Wed, 23 Aug 2023 18:52:10 +0800
Subject: [PATCH] Update config.py

---
 config.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/config.py b/config.py
index 53f8135..a5a0689 100644
--- a/config.py
+++ b/config.py
@@ -168,10 +168,11 @@ class Config:
                     )
                 except:
                     pass
-            import torch_directml
-
-            self.device = torch_directml.device(torch_directml.default_device())
-            self.is_half = False
+            if(self.device!="cpu"):
+                import torch_directml
+    
+                self.device = torch_directml.device(torch_directml.default_device())
+                self.is_half = False
         else:
             if self.instead:
                 print(f"use {self.instead} instead")