search

xlang-ai / instructor-embedding

[ACL 2023] One Embedder, Any Task: Instruction-Finetuned Text Embeddings
Apache License 2.0
1.85k stars 134 forks source link

NameError: name 'disabled_tqdm' is not defined #121

Open marybloodyzz opened 3 months ago

marybloodyzz commented 3 months ago

As the title, has anyone got similar issue? Traceback (most recent call last): File "/data//instructor-embedding/clustering.py", line 3, in model = Instructor('hkunlp/instructor-large') File "/home/miniconda3/envs/instructor/lib/python3.10/site-packages/sentence_transformers/SentenceTransformer.py", line 95, in init modules = self._load_sbert_model(model_path) File "/data/instructor-embedding/InstructorEmbedding/instructor.py", line 530, in _load_sbert_model "tqdm_class": disabled_tqdm, NameError: name 'disabled_tqdm' is not defined

FYI: def _load_sbert_model(self, model_path, token = None, cache_folder = None, revision = None, trust_remote_code = False): """ Loads a full sentence-transformers model """

Taken mostly from: https://github.com/UKPLab/sentence-transformers/blob/66e0ee30843dd411c64f37f65447bb38c7bf857a/sentence_transformers/util.py#L544

    download_kwargs = {
        "repo_id": model_path,
        "revision": revision,
        "library_name": "sentence-transformers",
        "token": token,
        "cache_dir": cache_folder,
        "tqdm_class": disabled_tqdm,   (NameError: name 'disabled_tqdm' is not defined)
    }
altria-zewei-wang commented 3 months ago

Facing same problem, fixed with installing sentence transformer 2.3.0 (idk if there will be any other side effect tho)