Skip to content

Commit 9b1cab5

Browse files
committed
fix(musa): enable backend autoload for training
1 parent 5355831 commit 9b1cab5

1 file changed

Lines changed: 1 addition & 2 deletions

File tree

tests/functional_tests/train/qwen3/conf/0_6b_musa.yaml

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -34,10 +34,9 @@ experiment:
3434
HYDRA_FULL_ERROR: 1
3535
FS_PLATFORM: musa
3636
MG_FL_PREFER: musa
37+
TORCH_FORCE_NO_WEIGHTS_ONLY_LOAD: 1
3738
# The pinned S5000 TransformerEngine build is validated in eager mode.
3839
# Disable torch/TE compilation to avoid unsupported Triton lowering paths.
39-
TORCH_DEVICE_BACKEND_AUTOLOAD: 0
40-
TORCH_FORCE_NO_WEIGHTS_ONLY_LOAD: 1
4140
TORCHDYNAMO_DISABLE: 1
4241
TORCH_COMPILE_DISABLE: 1
4342
NVTE_TORCH_COMPILE: 0

0 commit comments

Comments
 (0)