Fix Dockerfile.rocm (#2101)

Co-authored-by: miloice <jeffaw99@hotmail.com>
This commit is contained in:
TJian 2023-12-14 16:45:58 +08:00 committed by GitHub
parent 6774bd50b0
commit 05bdf4eaf3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -52,7 +52,7 @@ RUN pip install xformers==0.0.23 --no-deps
RUN cd /app \
&& cd vllm \
&& pip install -U -r requirements-rocm.txt \
&& bash patch_xformers-0.0.23.rocm.sh \
&& bash patch_xformers.rocm.sh \
&& python3 setup.py install \
&& cd ..