File tree Expand file tree Collapse file tree 1 file changed +19
-9
lines changed Expand file tree Collapse file tree 1 file changed +19
-9
lines changed Original file line number Diff line number Diff line change 23
23
required : true
24
24
type : string
25
25
description : " The distribution artifact name of torch_npu"
26
+ secrets :
27
+ hf_token :
28
+ description : ' A token passed from the caller workflow'
29
+ required : true
26
30
27
31
defaults :
28
32
run :
69
73
with :
70
74
repository : pytorch/torchtune
71
75
path : torchtune
76
+
77
+ - name : Install torchtune
78
+ working-directory : torchtune
79
+ run : |
80
+ pip install -e .
72
81
73
82
- name : Download torch artifact
74
83
if : ${{ inputs.torch-artifact }}
@@ -103,16 +112,19 @@ jobs:
103
112
working-directory : ascend_npu
104
113
run : |
105
114
pip install ${{ inputs.torch-npu-artifact }}
106
-
107
- - name : Install torchtune
108
- working-directory : torchtune
115
+
116
+ - name : Show environment info
109
117
run : |
110
- pip install -e .
111
-
118
+ pip list
119
+
112
120
- name : Download Qwen2.5 model
121
+ env :
122
+ HF_TOKEN : ${{ secrets.hf_token }}
113
123
run : |
114
- tune download Qwen/Qwen2.5-0.5B-Instruct \
115
- --output-dir /tmp/Qwen2.5-0.5B-Instruct
124
+ export HF_ENDPOINT=https://hf-mirror.com
125
+ huggingface-cli download --resume-download Qwen/Qwen2.5-0.5B-Instruct \
126
+ --local-dir /tmp/Qwen2.5-0.5B-Instruct \
127
+ --token $HF_TOKEN
116
128
117
129
- name : Run torchtune with lora finetune
118
130
run : |
@@ -121,5 +133,3 @@ jobs:
121
133
- name : Run torchtune with full finetune
122
134
run : |
123
135
tune run full_finetune_single_device --config qwen2_5/0.5B_full_single_device || true
124
-
125
-
You can’t perform that action at this time.
0 commit comments