Update config.py

2025-03-11 00:04:55 +08:00 · 2023-08-23 18:52:10 +08:00 · 2023-08-23 18:52:10 +08:00 · b4c10ce37d
commit b4c10ce37d
parent 2789b8e8f4
1 changed files with 5 additions and 4 deletions
--- a/config.py
+++ b/config.py
@ -168,6 +168,7 @@ class Config:
                    )
                except:
                    pass
+            if(self.device!="cpu"):
                import torch_directml
    
                self.device = torch_directml.device(torch_directml.default_device())