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,11 +181,11 @@ class Config:
                    )
                except:
                    pass
-            if self.device != "cpu":
-                import torch_directml
+            #if self.device != "cpu":
+            import torch_directml

-                self.device = torch_directml.device(torch_directml.default_device())
-                self.is_half = False
+            self.device = torch_directml.device(torch_directml.default_device())
+            self.is_half = False
        else:
            if self.instead:
                print(f"use {self.instead} instead")