fix(config): dml load & av codec ctx has no base_rate attr

Author: fumiama

configs/config.py

@@ -50,6 +50,7 @@ def __init__(self):
             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 @@ def device_config(self):
 
             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")

infer/lib/audio.py

@@ -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