Disable deepspeed

This commit is contained in:
Josh XT committed 2024-01-22 18:10:20 -05:00
1 parent 51c91c3d70
commit 360f70cb0a
2 files changed
+2 -3

No files matched your search

+1 -2
View File
@@ -13,7 +13,6 @@ ENV CUDA_DOCKER_ARCH=all
RUN pip install --no-cache-dir -r requirements_nvidia.txt && \
pip install --no-cache-dir -r requirements_finetune.txt && \
pip install --no-cache-dir -r requirements_other.txt && \
pip install --no-cache-dir deepspeed
pip install --no-cache-dir -r requirements_other.txt
EXPOSE 7851
CMD ["python", "script.py"]