diff --git a/configs/config.py b/configs/config.py index 6ca948e..ee120ee 100644 --- 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")