Disable deepspeed
This commit is contained in:
1 parent
51c91c3d70
commit
360f70cb0a
2 files changed
+2
-3
No files matched your search
+1
-2
@@ -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"]
|
||||
Reference in new issue
Block a user