Update README.md

README.md

@@ -262,6 +262,89 @@ huggingface-cli download lightx2v/Autoencoders \
     --local-dir ./models/vae/
 ```
 
+### 🧪  Video Reconstruction Test
+
+We provide a standalone script `vid_recon.py` to test VAE models independently. This script reads a video, encodes it through VAE, then decodes it back to verify the reconstruction quality.
+
+**Script Location**: `LightX2V/lightx2v/models/video_encoders/hf/vid_recon.py`
+
+```bash
+git clone https://github.com/ModelTC/LightX2V.git
+cd LightX2V
+```
+
+**1. Test Official VAE (Wan2.1)**
+```bash
+python -m lightx2v.models.video_encoders.hf.vid_recon \
+    input_video.mp4 \
+    --checkpoint ./models/vae/Wan2.1_VAE.pth \
+    --model_type vaew2_1 \
+    --device cuda \
+    --dtype bfloat16
+```
+
+**2. Test Official VAE (Wan2.2)**
+```bash
+python -m lightx2v.models.video_encoders.hf.vid_recon \
+    input_video.mp4 \
+    --checkpoint ./models/vae/Wan2.2_VAE.pth \
+    --model_type vaew2_2 \
+    --device cuda \
+    --dtype bfloat16
+```
+
+**3. Test LightTAE (Wan2.1)**
+```bash
+python -m lightx2v.models.video_encoders.hf.vid_recon \
+    input_video.mp4 \
+    --checkpoint ./models/vae/lighttaew2_1.pth \
+    --model_type taew2_1 \
+    --device cuda \
+    --dtype bfloat16
+```
+
+**4. Test LightTAE (Wan2.2)**
+```bash
+python -m lightx2v.models.video_encoders.hf.vid_recon \
+    input_video.mp4 \
+    --checkpoint ./models/vae/lighttaew2_2.pth \
+    --model_type taew2_2 \
+    --device cuda \
+    --dtype bfloat16
+```
+
+**5. Test LightVAE (Wan2.1)**
+```bash
+python -m lightx2v.models.video_encoders.hf.vid_recon \
+    input_video.mp4 \
+    --checkpoint ./models/vae/lightvaew2_1.pth \
+    --model_type vaew2_1 \
+    --device cuda \
+    --dtype bfloat16 \
+    --use_lightvae
+```
+
+
+**6. Test TAE (Wan2.1)**
+```bash
+python -m lightx2v.models.video_encoders.hf.vid_recon \
+    input_video.mp4 \
+    --checkpoint ./models/vae/taew2_1.pth \
+    --model_type taew2_1 \
+    --device cuda \
+    --dtype bfloat16
+```
+
+**7. Test TAE (Wan2.2)**
+```bash
+python -m lightx2v.models.video_encoders.hf.vid_recon \
+    input_video.mp4 \
+    --checkpoint ./models/vae/taew2_2.pth \
+    --model_type taew2_1 \
+    --device cuda \
+    --dtype bfloat16
+```
+
 ### Use in LightX2V
 
 Specify the VAE path in the configuration file: