Update config.py

This commit is contained in:
RVC-Boss 2023-08-30 18:57:40 +08:00 committed by GitHub
parent 8054281b24
commit 47148851f8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -181,11 +181,11 @@ 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())
self.is_half = False self.is_half = False
else: else:
if self.instead: if self.instead:
print(f"use {self.instead} instead") print(f"use {self.instead} instead")