Skip to content

Commit 0883811

Browse files
authored
Update _ascend_npu_torchtune.yml
1 parent dd9388d commit 0883811

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

.github/workflows/_ascend_npu_torchtune.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -108,7 +108,7 @@ jobs:
108108
working-directory: torchtune
109109
run: |
110110
# pip install -e .
111-
pip install torch==2.8.0 torchao torchvision
111+
pip install torch torchao torchvision --index-url https://download.pytorch.org/whl/nightly/cpu
112112
pip install torchtune
113113
114114
- name: Download Qwen2.5 model
@@ -124,4 +124,4 @@ jobs:
124124
run: |
125125
tune run full_finetune_single_device --config qwen2_5/0.5B_full_single_device || true
126126
127-
127+

0 commit comments

Comments
 (0)