Wing Lian 1302e31049 Transformers version flexibility and FSDP optimizer patch (#2155) ...

* allow flexibility in transformers version for FSDP

* more flexibility with dev versions of 4.47.0.dev0

* add patch for fsdp

* fix typo

* correct fn name

* stray character

* fix patch

* reset Trainer too

* also reset Trainer.training_step

* allow tests/patched to run more than one process on e2e runner

* skip tests/patched in e2e for now since it's run in regular pytest

2024-12-08 14:50:40 -05:00

..

Dockerfile

drop unnecessary BNB_CUDA_VERSION env var from docker as it just results in warnings (#2121) [skip ci]

2024-12-04 12:25:47 -05:00

Dockerfile-base

Transformers version flexibility and FSDP optimizer patch (#2155)

2024-12-08 14:50:40 -05:00

Dockerfile-cloud

Transformers 4.47.0 (#2138)

2024-12-07 05:03:01 -05:00

Dockerfile-cloud-no-tmux

Transformers 4.47.0 (#2138)

2024-12-07 05:03:01 -05:00

Dockerfile-tests

drop unnecessary BNB_CUDA_VERSION env var from docker as it just results in warnings (#2121) [skip ci]

2024-12-04 12:25:47 -05:00