fix: update to work with cache text encoder outputs (without disk)

2026-04-08 06:28:48 +00:00 · 2025-06-29 21:58:43 +09:00
parent 935e0037dc
commit 884c1f37c4
1 changed files with 1 additions and 1 deletions
--- 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,
                ]