Skip to content

Commit 55ef20e

Browse files
build(deps): upgrade transformers to 4.49 (#485)
* upgrade transformers to 4.49 Signed-off-by: Anh Uong <anh.uong@ibm.com> * remove test since output_dir not required anymore Signed-off-by: Anh Uong <anh.uong@ibm.com> --------- Signed-off-by: Anh Uong <anh.uong@ibm.com> Co-authored-by: Dushyant Behl <dushyantbehl@users.noreply.github.com>
1 parent 567034a commit 55ef20e

File tree

2 files changed

+1
-9
lines changed

2 files changed

+1
-9
lines changed

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ classifiers=[
2929
dependencies = [
3030
"numpy>=1.26.4,<2.0",
3131
"accelerate>=0.20.3,!=0.34,<1.1",
32-
"transformers>=4.46,<4.48.2",
32+
"transformers>=4.49,<5.0",
3333
"torch>=2.2.0,<2.5",
3434
"sentencepiece>=0.1.99,<0.3",
3535
"tokenizers>=0.13.3,<1.0",

tests/test_sft_trainer.py

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -322,14 +322,6 @@ def _get_training_logs_by_epoch(dir_path: str, epoch: int = None):
322322
return data_list
323323

324324

325-
def test_run_train_requires_output_dir():
326-
"""Check fails when output dir not provided."""
327-
updated_output_dir_train_args = copy.deepcopy(TRAIN_ARGS)
328-
updated_output_dir_train_args.output_dir = None
329-
with pytest.raises(TypeError):
330-
sft_trainer.train(MODEL_ARGS, DATA_ARGS, updated_output_dir_train_args, None)
331-
332-
333325
def test_run_train_fails_training_data_path_not_exist():
334326
"""Check fails when data path not found."""
335327
updated_data_path_args = copy.deepcopy(DATA_ARGS)

0 commit comments

Comments
 (0)