Update train_causal_video_vae.sh

scripts/train_causal_video_vae.sh

@@ -19,7 +19,7 @@ export HIP_VISIBLE_DEVICES=4,5,6,7
 
 GPUS=4  # The gpu number
 VAE_MODEL_PATH=/home/modelzoo/Pyramid-Flow/pyramid_flow_model/pyramid-flow-miniflux/causal_video_vae   # The vae model dir
-LPIPS_CKPT=/home/modelzoo/Pyramid-Flow/pyramid_flow_model/pyramid-flow-miniflux/vgg.pth    # The LPIPS VGG CKPT path, used for calculating the lpips loss
+LPIPS_CKPT=/home/modelzoo/Pyramid-Flow/pyramid_flow_model/pyramid-flow-miniflux/vgg_lpips.pth    # The LPIPS VGG CKPT path, used for calculating the lpips loss
 OUTPUT_DIR=./temp_vae    # The checkpoint saving dir
 IMAGE_ANNO=annotation/customs/vae_image.jsonl   # The image annotation file path
 VIDEO_ANNO=annotation/customs/vae_video.jsonl   # The video annotation file path