From f37fa88824db0428c67a85e8fbc8c7a359ef9a7a Mon Sep 17 00:00:00 2001 From: Disty0 Date: Thu, 25 Dec 2025 14:30:37 +0300 Subject: [PATCH] remove ModuleList and ModuleDict from no_split_module_classes --- modules/sd_offload.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/sd_offload.py b/modules/sd_offload.py index caafcc46a..846a38773 100644 --- a/modules/sd_offload.py +++ b/modules/sd_offload.py @@ -18,7 +18,7 @@ offload_warn = ['sc', 'sd3', 'f1', 'f2', 'h1', 'hunyuandit', 'auraflow', 'omnige offload_post = ['h1'] offload_hook_instance = None balanced_offload_exclude = ['CogView4Pipeline', 'MeissonicPipeline'] -no_split_module_classes = ["Linear", "Conv1d", "Conv2d", "Conv3d", "ConvTranspose1d", "ConvTranspose2d", "ConvTranspose3d", "WanTransformerBlock", "ModuleDict", "ModuleList"] +no_split_module_classes = ["Linear", "Conv1d", "Conv2d", "Conv3d", "ConvTranspose1d", "ConvTranspose2d", "ConvTranspose3d", "WanTransformerBlock"] accelerate_dtype_byte_size = None move_stream = None