diff --git a/Dockerfile.a100 b/Dockerfile.a100
index a13584e..852ec90 100644
--- a/Dockerfile.a100
+++ b/Dockerfile.a100
@@ -2,6 +2,7 @@ FROM pytorch/pytorch:2.6.0-cuda12.4-cudnn9-devel
 
 WORKDIR /workspace
 
-RUN pip install opencv-python-headless imageio[ffmpeg] einops datasets==3.2.0 simplejson diffusers==0.34.0 open_clip_torch==2.24.0 sortedcontainers modelscope av==11.0.0 addict -i https://nexus.4pd.io/repository/pypi-all/simple
+RUN pip install pytorch-lightning opencv-python-headless imageio[ffmpeg] einops datasets==3.2.0 simplejson diffusers==0.34.0 open_clip_torch==2.24.0 sortedcontainers modelscope av==11.0.0 addict -i https://nexus.4pd.io/repository/pypi-all/simple
+RUN pip install transformers accelerate -i https://nexus.4pd.io/repository/pypi-all/simple
 
 COPY . /workspace/
diff --git a/run_in_docker_a100.sh b/run_in_docker_a100.sh
index 01291cb..8919057 100755
--- a/run_in_docker_a100.sh
+++ b/run_in_docker_a100.sh
@@ -1,3 +1,3 @@
 #! /usr/bin/env bash
-image=harbor-contest.4pd.io/zhanghao/t2v:a100-0.1
+image=harbor-contest.4pd.io/zhanghao/t2v:a100-0.2
 docker run -it -v /home/zhanghao/workspace:/workspace -v /mnt:/mnt $image bash