This commit is contained in:
Your Name
2024-06-24 00:32:47 +00:00
parent 13ba53eaca
commit 0bcdcb72ef

View File

@@ -25,9 +25,6 @@ services:
TORCHVISION_VERSION: "0.18.0" TORCHVISION_VERSION: "0.18.0"
TORCHAUDIO_VERSION: "2.3.0" TORCHAUDIO_VERSION: "2.3.0"
DS_BUILD_OPS: 1 DS_BUILD_OPS: 1
# DS_BUILD_SPARSE_ATTN: 0
# DS_BUILD_FUSED_ADAM: 1
# DS_BUILD_CPU_ADAM: 1
USE_CUDA: 1 USE_CUDA: 1
USE_ROCM: 0 USE_ROCM: 0
USE_XPU: 0 USE_XPU: 0