instructor-embedding icon indicating copy to clipboard operation
instructor-embedding copied to clipboard

NameError: name 'disabled_tqdm' is not defined

Open marybloodyzz opened this issue 7 months ago • 1 comments

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) }

marybloodyzz avatar Jul 08 '24 17:07 marybloodyzz