KeyError: 'CodeXEmbedConfig'
When I used your example of sentence_transformers, the following error occurred. I downloaded the model to my local device and then loaded it from the local location. Could you please tell me where my mistake lies?
Traceback (most recent call last):
File "/media/dell/205datadisk_1/home/zcj/zcj_paper/dataset/bigvul/test.py", line 14, in
model = SentenceTransformer('/media/dell/205datadisk_1/home/zcj/zcj_paper/nano-graphrag/models/models--Salesforce--SFR-Embedding-Code-2B_R/snapshots/c73d8631a005876ed5abde34db514b1fb6566973', trust_remote_code=True)
File "/media/dell/205datadisk_1/home/zcj/miniconda3/envs/nanoG/lib/python3.10/site-packages/sentence_transformers/SentenceTransformer.py", line 287, in init
modules = self._load_sbert_model(
File "/media/dell/205datadisk_1/home/zcj/miniconda3/envs/nanoG/lib/python3.10/site-packages/sentence_transformers/SentenceTransformer.py", line 1487, in _load_sbert_model
module = Transformer(model_name_or_path, cache_dir=cache_folder, **kwargs)
File "/media/dell/205datadisk_1/home/zcj/miniconda3/envs/nanoG/lib/python3.10/site-packages/sentence_transformers/models/Transformer.py", line 54, in init
self._load_model(model_name_or_path, config, cache_dir, **model_args)
File "/media/dell/205datadisk_1/home/zcj/miniconda3/envs/nanoG/lib/python3.10/site-packages/sentence_transformers/models/Transformer.py", line 85, in _load_model
self.auto_model = AutoModel.from_pretrained(
File "/media/dell/205datadisk_1/home/zcj/miniconda3/envs/nanoG/lib/python3.10/site-packages/transformers/models/auto/auto_factory.py", line 559, in from_pretrained
return model_class.from_pretrained(
File "/media/dell/205datadisk_1/home/zcj/miniconda3/envs/nanoG/lib/python3.10/site-packages/transformers/modeling_utils.py", line 262, in _wrapper
return func(*args, **kwargs)
File "/media/dell/205datadisk_1/home/zcj/miniconda3/envs/nanoG/lib/python3.10/site-packages/transformers/modeling_utils.py", line 4185, in from_pretrained
model = cls(config, *model_args, **model_kwargs)
File "/media/dell/205datadisk_1/home/zcj/.cache/huggingface/modules/transformers_modules/c73d8631a005876ed5abde34db514b1fb6566973/modeling_gemma2.py", line 1347, in init
self.tokenizer = AutoTokenizer.from_pretrained(config._name_or_path, trust_remote_code=True, device_map="auto")
File "/media/dell/205datadisk_1/home/zcj/miniconda3/envs/nanoG/lib/python3.10/site-packages/transformers/models/auto/tokenization_auto.py", line 960, in from_pretrained
tokenizer_class_py, tokenizer_class_fast = TOKENIZER_MAPPING[type(config)]
File "/media/dell/205datadisk_1/home/zcj/miniconda3/envs/nanoG/lib/python3.10/site-packages/transformers/models/auto/auto_factory.py", line 760, in getitem
model_type = self._reverse_config_mapping[key.name]