Skip to content

02_mlflow_logging_inference.py fails to create Endpoint (cannot import name 'cached_download' from 'huggingface_hub') #108

@bryn-baritompa-db

Description

@bryn-baritompa-db

Issue

Running llm-models/embedding/bge/bge-m3/02_mlflow_logging_inference.py fails to create a Model Serving Endpoint.
The Endpoint log shows the following error:
ImportError: cannot import name 'cached_download' from 'huggingface_hub'

Cause

Breaking changes in huggingface_hub v0.26.0

Temporary workaround

Pin the versions as follows here

pip_requirements=[
      "mlflow==2.11.2",
      "sentence-transformers==2.7.0",
      "torch==2.2.1",
      "transformers==4.41.2",
      "huggingface-hub==0.24.6"
],

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions