This PR updates the huggingface_inference_toolkit wheel (we're using a wheel since the package is not publicly released yet i.e. not in PyPI), since as reported by Google, the artifact download from Google Cloud Storage (GCS) via the AIP_STORAGE_URI environment variable was not working.
Finally, this PR also updates the Dockerfile images so as to install the extras from the wheel instead of installing the dependencies and then the wheel, so that the dependency management is defined within huggingface_inference_toolkit instead.
Description
This PR updates the
huggingface_inference_toolkitwheel (we're using a wheel since the package is not publicly released yet i.e. not in PyPI), since as reported by Google, the artifact download from Google Cloud Storage (GCS) via theAIP_STORAGE_URIenvironment variable was not working.So on, that issue was fixed at https://github.com/huggingface/huggingface-inference-toolkit/pull/76; and besides that we also included the
googleextra withinhuggingface_inference_toolkitso that thegoogle-cloud-storageinstallation (and any other required package in the future) is installed via the extras, see https://github.com/huggingface/huggingface-inference-toolkit/pull/77.Finally, this PR also updates the
Dockerfileimages so as to install the extras from the wheel instead of installing the dependencies and then the wheel, so that the dependency management is defined withinhuggingface_inference_toolkitinstead.