[Bugfix] Update Dockerfile.cpu to fix NameError: name 'vllm_ops' is not defined (#5009)

2024-05-23 17:08:58 +01:00
parent 6066253296
commit 2ba80bed27
2 changed files with 3 additions and 1 deletions
--- a/Dockerfile.cpu
+++ b/Dockerfile.cpu
@@ -17,4 +17,6 @@ RUN pip install -v -r requirements-cpu.txt --extra-index-url https://download.py

 RUN VLLM_TARGET_DEVICE=cpu python3 setup.py install

+WORKDIR /workspace/
+
 CMD ["/bin/bash"]