HuggingFace_transformer

Files

Ella Charlaix 39acfe84ba Add deepspeed test to amd scheduled CI (#27633 )

* add deepspeed scheduled test for amd

* fix image

* add dockerfile

* add comment

* enable tests

* trigger

* remove trigger for this branch

* trigger

* change runner env to trigger the docker build image test

* use new docker image

* remove test suffix from docker image tag

* replace test docker image with original image

* push new image

* Trigger

* add back amd tests

* fix typo

* add amd tests back

* fix

* comment until docker image build scheduled test fix

* remove deprecated deepspeed build option

* upgrade torch

* update docker & make tests pass

* Update docker/transformers-pytorch-deepspeed-amd-gpu/Dockerfile

* fix

* tmp disable test

* precompile deepspeed to avoid timeout during tests

* fix comment

* trigger deepspeed tests with new image

* comment tests

* trigger

* add sklearn dependency to fix slow tests

* enable back other tests

* final update

---------

Co-authored-by: Felix Marty <felix@hf.co>
Co-authored-by: Félix Marty <9808326+fxmarty@users.noreply.github.com>
Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>

2023-12-11 16:33:36 +01:00

transformers-all-latest-gpu

Faster generation using AWQ + Fused modules (#27411 )

2023-12-05 12:14:45 +01:00

transformers-cpu

TF: TF 2.10 unpin + related onnx test skips (#18995 )

2022-09-12 19:30:27 +01:00

transformers-doc-builder

Don't install pytorch-quantization in Doc Builder docker file (#26622 )

2023-10-05 16:57:50 +02:00

transformers-gpu

TF: TF 2.10 unpin + related onnx test skips (#18995 )