This commit is contained in:
2024-06-21 14:31:53 +08:00
parent 6bd77c71ca
commit d351fe520a
6 changed files with 400 additions and 7 deletions

View File

@@ -12,7 +12,7 @@ services:
TORCHVISION_VERSION: 0.14.1
TORCHAUDIO_VERSION: 0.13.1
DS_BUILD_OPS: 1
DS_BUILD_SPARSE_ATTN: 0
DS_BUILD_SPARSE_ATTN: 1
DS_BUILD_FUSED_ADAM: 1
DS_BUILD_CPU_ADAM: 1
USE_CUDA: 1