Add kernels==0.16.0 (trl 1.15.0 loads kernels-community/flash-attn3)
Browse files- run_job.sh +1 -0
run_job.sh
CHANGED
|
@@ -32,6 +32,7 @@ python3 -m pip install --quiet \
|
|
| 32 |
'trl[peft]==1.15.0' \
|
| 33 |
'openenv[harbor]' \
|
| 34 |
'harbor[hf-sandbox]>=0.24.0' \
|
|
|
|
| 35 |
math-verify tomli-w \
|
| 36 |
aiohttp
|
| 37 |
|
|
|
|
| 32 |
'trl[peft]==1.15.0' \
|
| 33 |
'openenv[harbor]' \
|
| 34 |
'harbor[hf-sandbox]>=0.24.0' \
|
| 35 |
+
'kernels==0.16.0' \
|
| 36 |
math-verify tomli-w \
|
| 37 |
aiohttp
|
| 38 |
|