search

huggingface / huggingface_hub

The official Python client for the Huggingface Hub.
https://huggingface.co/docs/huggingface_hub
Apache License 2.0
1.83k stars 471 forks source link

Support custom_image in update_inference_endpoint #2306

Closed Wauplin closed 1 month ago

Wauplin commented 1 month ago

Fix https://github.com/huggingface/huggingface_hub/issues/2301.

With create_inference_endpoint, it is possible to set a custom docker image to run in the Inference Endpoint (typically a TGI container). When adding this in https://github.com/huggingface/huggingface_hub/issues/1861, I forgot to add it to update_inference_endpoint. This PR fixes this for consistency. And also to save poor @datavistics regularly making bad dreams about it.

HuggingFaceDocBuilderDev commented 1 month ago

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.