Update config.py
This commit is contained in:
parent
8054281b24
commit
47148851f8
@ -181,7 +181,7 @@ class Config:
|
|||||||
)
|
)
|
||||||
except:
|
except:
|
||||||
pass
|
pass
|
||||||
if self.device != "cpu":
|
#if self.device != "cpu":
|
||||||
import torch_directml
|
import torch_directml
|
||||||
|
|
||||||
self.device = torch_directml.device(torch_directml.default_device())
|
self.device = torch_directml.device(torch_directml.default_device())
|
||||||
|
Loading…
x
Reference in New Issue
Block a user