From 4a5509a4c5e0518242b3f7a30577e4e0e4c63687 Mon Sep 17 00:00:00 2001 From: Yousef Rafat <81116377+yousef-rafat@users.noreply.github.com> Date: Sun, 16 Nov 2025 16:20:35 +0200 Subject: [PATCH] . --- comfy/ldm/hunyuan_image_3/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy/ldm/hunyuan_image_3/model.py b/comfy/ldm/hunyuan_image_3/model.py index 60ab43340..6b292ed7f 100644 --- a/comfy/ldm/hunyuan_image_3/model.py +++ b/comfy/ldm/hunyuan_image_3/model.py @@ -909,7 +909,7 @@ class HunyuanImage3Model(nn.Module): experts = [LazyMoELoader(self.moe_lru, self.config) for _ in range(64)] self.layers[layer_idx+1].mlp.experts = [expert._schedule_disk_load(layer_idx+1, i) for i, expert in enumerate(experts)] - if not additional_layers_set: + if not self.additional_layers_set: if (layer_idx % sparse_interval == 0) and layer_idx >= sparse_interval: experts = [LazyMoELoader(self.moe_lru, self.config) for _ in range(64)] self.layers[next_layers].mlp.experts = [expert._schedule_disk_load(next_layers, i) for i, expert in enumerate(experts)]