diff --git a/docker/pyt_hy_video.ubuntu.amd.Dockerfile b/docker/pyt_hy_video.ubuntu.amd.Dockerfile
index 062fc9fe..1f63d1a9 100644
--- a/docker/pyt_hy_video.ubuntu.amd.Dockerfile
+++ b/docker/pyt_hy_video.ubuntu.amd.Dockerfile
@@ -6,10 +6,11 @@ ARG work_dir=/hunyuanvideo
 
 WORKDIR $work_dir
 
-RUN pip install diffusers==0.32.2 distvae yunchang==0.6.0 opencv-python accelerate
+#RUN pip install diffusers==0.32.2 distvae yunchang==0.6.0 opencv-python accelerate
+RUN pip install transformers==4.56.2 diffusers==0.32.2 distvae yunchang==0.6.0 opencv-python-headless accelerate
 RUN pip install imageio imageio-ffmpeg
 RUN pip install beautifulsoup4==4.12.3
-RUN pip install sentencepiece==0.1.99
+RUN pip install sentencepiece>=0.2.0
 RUN pip install numpy==1.26.4
 
 # xDiT repository
@@ -19,7 +20,7 @@ RUN git clone https://github.com/xdit-project/xDiT && \
     pip install --no-deps -e .
 
 # flash attn
-ARG FA_SHA="22c0358"
+ARG FA_SHA="83f9e450cd10e20701fb109db9c7703d376f282b"
 ARG FA_REPO="https://github.com/ROCm/flash-attention.git"
 ARG PYTORCH_ROCM_ARCH="gfx942"
 RUN git clone ${FA_REPO} && \
@@ -28,12 +29,32 @@ RUN git clone ${FA_REPO} && \
     git submodule update --init && \
     GPU_ARCHS=${PYTORCH_ROCM_ARCH} python3 setup.py bdist_wheel --dist-dir=dist && \
     pip install dist/*.whl;
+RUN apt-get update && apt-get install -y \
+    build-essential \
+    cmake \
+    pkg-config \
+    xxd \
+    git \
+    python3-dev \
+    && rm -rf /var/lib/apt/lists/*
+
+# RPD profiling
+#RUN apt update && \
+#    apt install -y sqlite3 libsqlite3-dev libfmt-dev
+#RUN git clone https://github.com/ROCmSoftwarePlatform/rocmProfileData && \
+#    cd rocmProfileData && \
+#    make && make install && \
+#    cd rocpd_python && python setup.py install && cd .. && \
+#    cd rpd_tracer && python setup.py install && cd ..
+
 
 # RPD profiling
 RUN apt update && \
     apt install -y sqlite3 libsqlite3-dev libfmt-dev
 RUN git clone https://github.com/ROCmSoftwarePlatform/rocmProfileData && \
     cd rocmProfileData && \
-    make && make install && \
-    cd rocpd_python && python setup.py install && cd .. && \
-    cd rpd_tracer && python setup.py install && cd ..
+    make && \
+    make install -C rocpd_python && \
+    make install -C rpd_tracer && \
+    ldconfig
+                                           
diff --git a/docker/pyt_mochi_inference.ubuntu.amd.Dockerfile b/docker/pyt_mochi_inference.ubuntu.amd.Dockerfile
index 4cbbd1ef..fb9426f2 100644
--- a/docker/pyt_mochi_inference.ubuntu.amd.Dockerfile
+++ b/docker/pyt_mochi_inference.ubuntu.amd.Dockerfile
@@ -33,7 +33,7 @@ RUN mkdir -p $WORKSPACE_DIR
 WORKDIR $WORKSPACE_DIR
 
 ARG FA_REPO="https://github.com/Dao-AILab/flash-attention.git"
-ARG PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx1100;gfx1101;gfx1200;gfx1201
+ARG PYTORCH_ROCM_ARCH=gfx950;gfx90a;gfx942;gfx1100;gfx1101;gfx1200;gfx1201
 RUN git clone ${FA_REPO}
 RUN cd flash-attention \
     && git submodule update --init \