Update config.py

2025-03-11 08:15:01 +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())