mirror of
https://github.com/fumiama/Retrieval-based-Voice-Conversion-WebUI.git
synced 2026-06-05 01:10:22 +08:00
fix(config): dml load & av codec ctx has no base_rate attr
This commit is contained in:
@@ -50,6 +50,7 @@ class Config(metaclass=Singleton):
|
||||
self.nocheck,
|
||||
self.update,
|
||||
) = self.arg_parse()
|
||||
self.dml = False
|
||||
self.instead = ""
|
||||
self.preprocess_per = 3.7
|
||||
self.x_pad, self.x_query, self.x_center, self.x_max = self.device_config()
|
||||
@@ -239,6 +240,7 @@ class Config(metaclass=Singleton):
|
||||
|
||||
self.device = torch_directml.device(torch_directml.default_device())
|
||||
self.is_half = False
|
||||
self.dml = True
|
||||
else:
|
||||
if self.instead:
|
||||
logger.info(f"Use {self.instead} instead")
|
||||
|
||||
@@ -195,7 +195,10 @@ def get_audio_properties(input_path: str) -> Tuple[int, int]:
|
||||
container = av.open(input_path)
|
||||
audio_stream = next(s for s in container.streams if s.type == "audio")
|
||||
channels = 1 if audio_stream.layout == "mono" else 2
|
||||
rate = audio_stream.base_rate
|
||||
try:
|
||||
rate = audio_stream.base_rate
|
||||
except:
|
||||
rate = audio_stream.sample_rate
|
||||
container.close()
|
||||
return channels, rate
|
||||
|
||||
|
||||
Reference in New Issue
Block a user