From 47148851f8b5a35515c67a56b2a1813bf9a56e5b Mon Sep 17 00:00:00 2001 From: RVC-Boss <129054828+RVC-Boss@users.noreply.github.com> Date: Wed, 30 Aug 2023 18:57:40 +0800 Subject: [PATCH] Update config.py --- configs/config.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) 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")