From 5f4020b12d12983eba73ceec59755b56c396ab5d Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Fri, 31 Oct 2025 17:24:24 +0200
Subject: [PATCH] Fix a possible issue with Ovi audio model loading

---
 nodes_sampler.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/nodes_sampler.py b/nodes_sampler.py
index 6bf72b8..bcc7dd7 100644
--- a/nodes_sampler.py
+++ b/nodes_sampler.py
@@ -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