fix mtx-game model (#523)

lightx2v/models/networks/wan/infer/matrix_game2/pre_infer.py

 import torch
 
-from lightx2v.models.networks.wan.infer.module_io import GridOutput, WanPreInferModuleOutput
-from lightx2v.models.networks.wan.infer.self_forcing.pre_infer import WanSFPreInfer, sinusoidal_embedding_1d
+from lightx2v.models.networks.wan.infer.module_io import GridOutput
+from lightx2v.models.networks.wan.infer.self_forcing.pre_infer import WanSFPreInfer, WanSFPreInferModuleOutput, sinusoidal_embedding_1d
 from lightx2v.utils.envs import *
 
 
@@ -86,7 +86,7 @@ class WanMtxg2PreInfer(WanSFPreInfer):
         context_lens = None
         context = self.img_emb(weights, visual_context)
 
-        return WanPreInferModuleOutput(
+        return WanSFPreInferModuleOutput(
             embed=embed,
             grid_sizes=grid_sizes,
             x=x.squeeze(0),

test_cases/run_qwen_image_i2i_2509.sh

 #!/bin/bash
-export CUDA_VISIBLE_DEVICES=
 
 # set path and first
 export lightx2v_path=
 export model_path=
 
-# check section
-if [ -z "${CUDA_VISIBLE_DEVICES}" ]; then
-    cuda_devices=0
-    echo "Warn: CUDA_VISIBLE_DEVICES is not set, using default value: ${cuda_devices}, change at shell script or set env variable."
-    export CUDA_VISIBLE_DEVICES=${cuda_devices}
-fi
-
-if [ -z "${lightx2v_path}" ]; then
-    echo "Error: lightx2v_path is not set. Please set this variable first."
-    exit 1
-fi
-
-if [ -z "${model_path}" ]; then
-    echo "Error: model_path is not set. Please set this variable first."
-    exit 1
-fi
-
-export TOKENIZERS_PARALLELISM=false
-
-export PYTHONPATH=${lightx2v_path}:$PYTHONPATH
-export DTYPE=BF16
-export PROFILING_DEBUG_LEVEL=2
+export CUDA_VISIBLE_DEVICES=0
 
+# set environment variables
+source ${lightx2v_path}/scripts/base/base.sh
 
 python -m lightx2v.infer \
     --model_cls qwen_image \