зеркало из https://github.com/microsoft/DeepSpeed.git
Pin transformers to 4.45.2 in nv-ds-chat workflow (#6710)
This commit causes breaking changes we need to fix, for now we will pin the version but we will fix shortly https://github.com/huggingface/transformers/pull/33325
This commit is contained in:
Родитель
9068acb6fb
Коммит
6c08b7f932
|
@ -12,6 +12,7 @@ on:
|
|||
type: string
|
||||
pull_request:
|
||||
paths:
|
||||
- ".github/workflows/nv-ds-chat.yml"
|
||||
- "deepspeed/runtime/zero/stage_1_and_2.py"
|
||||
- "deepspeed/runtime/zero/stage3.py"
|
||||
- "deepspeed/runtime/hybrid_engine.py"
|
||||
|
@ -42,6 +43,7 @@ jobs:
|
|||
|
||||
- name: Install deepspeed
|
||||
run: |
|
||||
pip install transformers==4.45.2
|
||||
pip install .[dev]
|
||||
ds_report
|
||||
|
||||
|
|
Загрузка…
Ссылка в новой задаче