This commit is contained in:
2024-07-13 16:18:53 +08:00
parent 52d9b541a8
commit 1d0a7f211d

View File

@@ -34,8 +34,8 @@ services:
DCUTLASS_NVCC_ARCHS: "80;86;89;90;90a" # 90a for H100 GPU 89:GeForce RTX 4090 DCUTLASS_NVCC_ARCHS: "80;86;89;90;90a" # 90a for H100 GPU 89:GeForce RTX 4090
DEEPSPEED_VERSION: "master" DEEPSPEED_VERSION: "master"
DEEPSPEED_INSTALL_FLAGS: "--allow_sudo" DEEPSPEED_INSTALL_FLAGS: "--allow_sudo"
HTTP_PROXY: "socks5://127.0.0.1:40000" HTTP_PROXY: "http://127.0.0.1:15777"
HTTPS_PROXY: "socks5://127.0.0.1:40000" HTTPS_PROXY: "http://127.0.0.1:15777"
volumes: volumes:
- ./src:/bbtft - ./src:/bbtft
container_name: ubuntu-finetune container_name: ubuntu-finetune