diff --git a/finetune/docker-compose_pytorch2.3.yml b/finetune/docker-compose_pytorch2.3.yml index fa0a82b..4390e55 100644 --- a/finetune/docker-compose_pytorch2.3.yml +++ b/finetune/docker-compose_pytorch2.3.yml @@ -34,8 +34,8 @@ services: DCUTLASS_NVCC_ARCHS: "80;86;89;90;90a" # 90a for H100 GPU 89:GeForce RTX 4090 DEEPSPEED_VERSION: "master" DEEPSPEED_INSTALL_FLAGS: "--allow_sudo" - HTTP_PROXY: "socks5://127.0.0.1:40000" - HTTPS_PROXY: "socks5://127.0.0.1:40000" + HTTP_PROXY: "http://127.0.0.1:15777" + HTTPS_PROXY: "http://127.0.0.1:15777" volumes: - ./src:/bbtft container_name: ubuntu-finetune