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:
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")