[diffusion] hardware: support diffusion (single GPU, 3/N) (#17105)

Signed-off-by: Xiaodong Ye <xiaodong.ye@mthreads.com>

docker/rocm.Dockerfile

@@ -184,9 +184,9 @@ RUN git clone ${SGL_REPO} \
     && cd .. \
     && rm -rf python/pyproject.toml && mv python/pyproject_other.toml python/pyproject.toml \
     && if [ "$BUILD_TYPE" = "srt" ]; then \
-         python -m pip --no-cache-dir install -e "python[srt_hip,diffusion]"; \
+         python -m pip --no-cache-dir install -e "python[srt_hip,diffusion_hip]"; \
        else \
-         python -m pip --no-cache-dir install -e "python[all_hip,diffusion]"; \
+         python -m pip --no-cache-dir install -e "python[all_hip,diffusion_hip]"; \
        fi
 
 RUN python -m pip cache purge