From 884c1f37c4c16fa83ed14f46f6e209770fbed4d8 Mon Sep 17 00:00:00 2001 From: Kohya S Date: Sun, 29 Jun 2025 21:58:43 +0900 Subject: [PATCH] fix: update to work with cache text encoder outputs (without disk) --- library/strategy_lumina.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/library/strategy_lumina.py b/library/strategy_lumina.py index 3d86dbef..392d6594 100644 --- a/library/strategy_lumina.py +++ b/library/strategy_lumina.py @@ -264,8 +264,8 @@ class LuminaTextEncoderOutputsCachingStrategy(TextEncoderOutputsCachingStrategy) else: info.text_encoder_outputs = [ hidden_state_i, - attention_mask_i, input_ids_i, + attention_mask_i, ]