Update config.py

2023-08-23 18:52:10 +08:00 · 2023-08-23 18:52:10 +08:00 · b4c10ce37d
parent 2789b8e8f4
commit b4c10ce37d
1 changed files with 5 additions and 4 deletions
--- 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")