From c1c4d6e0eefc3dd1260ce6ce25f4e2f647ecefe0 Mon Sep 17 00:00:00 2001
From: Ryan Meador <1176600+ryanmeador@users.noreply.github.com>
Date: Wed, 7 Jan 2026 22:19:11 -0500
Subject: [PATCH] prevent accelerate from recreating keys such as  and  so that
 the VAE decode will progress. resolves #4526

---
 modules/sd_offload.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/modules/sd_offload.py b/modules/sd_offload.py
index 846a38773..819f360d1 100644
--- a/modules/sd_offload.py
+++ b/modules/sd_offload.py
@@ -254,10 +254,12 @@ class OffloadHook(accelerate.hooks.ModelHook):
             if debug:
                 shared.log.trace(f'Offload: type=balanced op=dispatch map={device_map}')
             if device_map is not None:
+                skip_keys = getattr(module, "_skip_keys", None)
                 module = accelerate.dispatch_model(module,
                                                    main_device=torch.device(devices.device),
                                                    device_map=device_map,
                                                    offload_dir=offload_dir,
+                                                   skip_keys=skip_keys,
                                                    force_hooks=True,
                                                   )
             module._hf_hook.execution_device = torch.device(devices.device) # pylint: disable=protected-access