fix: improved regex for patching relative imports in bundled VibeVoice modules (v1.5.00)

2025-12-29 14:08:30 +08:00
parent 9a5875d9d7
commit 7e02593cf7
1 changed files with 4 additions and 1 deletions
@@ -99,7 +99,10 @@ class AIIA_VibeVoice_Loader:
                    source_code = f.read()
                
                # PATCH: Convert relative imports to absolute
-                source_code = re.sub(r'from \.(\w+)', r'from \1', source_code)
+                # Regex handles "from .module" with variable spacing
+                source_code, n_subs = re.subn(r'from\s+\.(\w+)', r'from \1', source_code)
+                if n_subs > 0:
+                    print(f"[AIIA] Patched {n_subs} relative imports in {module_name}")

                # PATCH: Fix unsafe .to(device) on potential None types in VibeVoice generation code
                # (Ideally this should be fixed in the source file, but we keep this just in case users load unpatched files)