423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-14 19:06:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add ARG HTTP_PROXY in Dockerfile to support HTTP proxy during image building.

Browse Source 
Former-commit-id: f1d76786e0
This commit is contained in:
shibingli@yeah.net
2024-12-27 18:31:14 +08:00
parent a37ef0eaae
commit c76c33ddb1
3 changed files with 58 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docker/docker-cuda/Dockerfile
View File

@@ -22,6 +22,7 @@ ARG HTTP_PROXY=
# Set the working directory
WORKDIR /app
# Set http proxy
RUN if [ -n "$HTTP_PROXY" ]; then \
echo "Configuring proxy..."; \
export http_proxy=$HTTP_PROXY; \
@@ -82,6 +83,7 @@ RUN pip uninstall -y transformer-engine flash-attn && \
fi
# Unset http proxy
RUN if [ -n "$HTTP_PROXY" ]; then \
unset http_proxy; \
unset https_proxy; \