2025-01-01 04:06:11.065139: E external/local_xla/xla/stream_executor/cuda/cuda_fft.cc:485] Unable to register cuFFT factory: Attempting to register factory for plugin cuFFT when one has already been registered
2025-01-01 04:06:11.078453: E external/local_xla/xla/stream_executor/cuda/cuda_dnn.cc:8454] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered
2025-01-01 04:06:11.082300: E external/local_xla/xla/stream_executor/cuda/cuda_blas.cc:1452] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered
INFO:mteb.cli:Running with parameters: Namespace(model='avsolatorio/NoInstruct-small-Embedding-v0', task_types=None, categories=None, tasks=['BiorxivClusteringP2P.v2'], languages=None, benchmarks=None, device=None, output_folder='/data/niklas/results/results', verbosity=2, co2_tracker=True, eval_splits=None, model_revision=None, batch_size=32, overwrite=False, save_predictions=False, func=<function run at 0x7f7927450f70>)
WARNING:mteb.model_meta:Loader not specified for model avsolatorio/NoInstruct-small-Embedding-v0, loading using sentence transformers.
Traceback (most recent call last):
File "/env/lib/conda/gritkto/bin/mteb", line 8, in
sys.exit(main())
File "/data/niklas/mteb/mteb/cli.py", line 387, in main
args.func(args)
File "/data/niklas/mteb/mteb/cli.py", line 123, in run
model = mteb.get_model(args.model, args.model_revision, device=device, trust_remote_code=True)
File "/data/niklas/mteb/mteb/models/overview.py", line 150, in get_model
model = meta.load_model(**kwargs)
File "/data/niklas/mteb/mteb/model_meta.py", line 128, in load_model
model: Encoder = loader(**kwargs) # type: ignore
File "/data/niklas/mteb/mteb/model_meta.py", line 40, in sentence_transformers_loader
return SentenceTransformerWrapper(model=model_name, revision=revision, **kwargs)
File "/data/niklas/mteb/mteb/models/sentence_transformer_wrapper.py", line 38, in init
self.model = SentenceTransformer(model, revision=revision, **kwargs)
File "/env/lib/conda/gritkto/lib/python3.10/site-packages/sentence_transformers/SentenceTransformer.py", line 306, in init
modules, self.module_kwargs = self._load_sbert_model(
File "/env/lib/conda/gritkto/lib/python3.10/site-packages/sentence_transformers/SentenceTransformer.py", line 1738, in _load_sbert_model
module = module_class.load(module_path)
File "/env/lib/conda/gritkto/lib/python3.10/site-packages/sentence_transformers/models/Pooling.py", line 247, in load
return Pooling(**config)
TypeError: Pooling.init() got an unexpected keyword argument 'output_key'