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,
                 ]