huggingface / optimum-tpu

Google TPU optimizations for transformers models

Apache License 2.0

76 stars 19 forks source link

fix(docker): correct jetstream installation in TGI docker image #107

Closed tengomucho closed 1 month ago

tengomucho commented 1 month ago

What does this PR do?

Fix yet another place where the Jetstream Pytorch installation was broken.