Update config.py

2023-08-30 18:57:40 +08:00 · 2023-08-30 18:57:40 +08:00 · 47148851f8
parent 8054281b24
commit 47148851f8
1 changed files with 4 additions and 4 deletions
--- a/configs/config.py
+++ b/configs/config.py
@ -181,7 +181,7 @@ class Config:
                    )
                except:
                    pass
-            if self.device != "cpu":
+            #if self.device != "cpu":
            import torch_directml

            self.device = torch_directml.device(torch_directml.default_device())