ci(docker): update torch-xla to 2.4.0

huggingface / optimum-tpu

Google TPU optimizations for transformers models

Apache License 2.0

75 stars 19 forks source link

ci(docker): update torch-xla to 2.4.0 #89

Closed tengomucho closed 2 months ago

tengomucho commented 2 months ago

Now it has been released, it makes sense to use that version. Without this, the CI was running the (moderate) risk of using an unstable/unreleased/unsupported version of torch and torch xla.