Fix a possible issue with Ovi audio model loading

2025-10-31 17:24:24 +02:00
parent 5da8a6b169
commit 5f4020b12d
1 changed files with 5 additions and 0 deletions
@@ -189,6 +189,11 @@ class WanVideoSampler:
        vae_upscale_factor = 16 if is_5b else 8

        # Load weights
+        if transformer.audio_model is not None:
+            for block in transformer.blocks:
+                if hasattr(block, 'audio_block'):
+                    block.audio_block = None
+
        if not transformer.patched_linear and patcher.model["sd"] is not None and len(patcher.patches) != 0:
            transformer = _replace_linear(transformer, dtype, patcher.model["sd"])
            transformer.patched_linear = True