diff --git a/examples/contrastive-image-text/README.md b/examples/contrastive-image-text/README.md index b526c6091b..cd4aa92295 100644 --- a/examples/contrastive-image-text/README.md +++ b/examples/contrastive-image-text/README.md @@ -250,5 +250,8 @@ python run_clip.py \ --use_lazy_mode \ --use_hpu_graphs_for_inference \ --gaudi_config_name Habana/clip \ - --bf16 + --bf16 \ + --mediapipe_dataloader ``` + +> `--mediapipe_dataloader` only works on Gaudi2. diff --git a/optimum/habana/version.py b/optimum/habana/version.py index 5ea2b1d648..46b25684dd 100644 --- a/optimum/habana/version.py +++ b/optimum/habana/version.py @@ -13,4 +13,4 @@ # See the License for the specific language governing permissions and # limitations under the License. -__version__ = "1.10.0.dev0" +__version__ = "1.10.0" diff --git a/setup.py b/setup.py index 100348d060..3ab10a2220 100644 --- a/setup.py +++ b/setup.py @@ -49,7 +49,7 @@ QUALITY_REQUIRES = [ "ruff", - "hf_doc_builder @ git+https://github.com/huggingface/doc-builder.git", + "hf_doc_builder", ] EXTRAS_REQUIRE = {