more graceful log and shutdown on unified script

2026-01-13 00:44:11 +02:00
parent 346f9ccbda
commit 1b54e4d5e2
4 changed files with 8 additions and 1 deletions
--- a/headless_rvc.py
+++ b/headless_rvc.py
@@ -241,7 +241,14 @@ class HeadlessRVC:
        finally:
            os.chdir(saved_cwd)
        
-        self.tgt_sr = self.rvc.tgt_sr
+        # Get target sample rate (with fallback)
+        if hasattr(self.rvc, 'tgt_sr') and self.rvc.tgt_sr:
+            self.tgt_sr = self.rvc.tgt_sr
+        else:
+            # Fallback to config sample rate if tgt_sr not set
+            logger.warning("RVC tgt_sr not set, using config sample rate")
+            self.tgt_sr = self.gui_config.samplerate
+            
        self.gui_config.samplerate = (
            self.tgt_sr if self.tgt_sr else self.gui_config.samplerate
        )