search

huggingface / optimum-benchmark

🏋️ A unified multi-backend utility for benchmarking Transformers, Timm, PEFT, Diffusers and Sentence-Transformers with full support of Optimum's hardware optimizations & quantization schemes.
Apache License 2.0
255 stars 48 forks source link

fix broken cuda and rocm images #263

Closed baptistecolle closed 2 months ago

baptistecolle commented 2 months ago

AutoAWQ change its setup.py, this broke the building of our docker image

IlyasMoutawwakil commented 2 months ago

can you fix the same TORCH_VERSION issue in rocm image as well (we build the same packages there)

baptistecolle commented 2 months ago

done, rocm is fixed

IlyasMoutawwakil commented 2 months ago

I'm not sure about that (see failing PyTorch tests), I think https://github.com/casper-hansen/AutoAWQ/blob/main/setup.py#L10 is triggering the installation of ipex https://github.com/huggingface/optimum-benchmark/actions/runs/10908257743/job/30273811027?pr=263#step:5:129

baptistecolle commented 2 months ago

Now, everything concerning the broken Docker image due to an update with AutoAWQ should have been fixed.

There are still some failing tests in our pipeline at of today: (because of #265 - issue with the codecarbon lock)