MCPcopy Create free account

hub / github.com/QwenLM/Qwen3-Embedding / functions

Functions79 in github.com/QwenLM/Qwen3-Embedding

↓ 5 callersMethodencode
(self, sentences: Union[List[str], str], is_query: bool = False, instruction=None, dim: int = -1)
examples/qwen3_embedding_vllm.py:20
↓ 4 callersMethodembed
( self, sentences: Sequence[str], max_length: int, prompt: str | None = None
evaluation/qwen3_embedding_model.py:57
↓ 2 callersMethodencode
(self, sentences: Union[List[str], str], is_query: bool = False, instruction=None, dim: int = -1)
examples/qwen3_embedding_transformers.py:43
↓ 2 callersFunctionload_belebel_data
In this retrieval datasets, corpus is in lang XX and queries in lang YY.
evaluation/utils.py:75
↓ 2 callersFunctionload_hagrid_data
(self, **kwargs)
evaluation/utils.py:104
↓ 2 callersFunctionload_mlqa_data
In this retrieval datasets, corpus is in lang XX and queries in lang YY.
evaluation/utils.py:50
↓ 2 callersFunctionload_rarb_data
(self, **kwargs)
evaluation/utils.py:11
↓ 2 callersFunctionrun_bright
(t, model, args, **kwargs)
evaluation/run_mteb.py:254
↓ 2 callersMethodstart
(self)
evaluation/qwen3_embedding_model.py:315
↓ 2 callersMethodstop
(self)
evaluation/qwen3_embedding_model.py:333
↓ 1 callersMethod__init__
( self, model: str, use_instruction: bool = False, device: str = 'cuda',
evaluation/qwen3_embedding_model.py:169
↓ 1 callersMethodcompute_logits
(self, inputs, **kwargs)
examples/qwen3_reranker_transformers.py:72
↓ 1 callersMethodcompute_scores
(self, pairs, **kwargs)
examples/qwen3_reranker_vllm.py:62
↓ 1 callersMethodcompute_scores
( self, pairs, instruction=None, **kwargs )
examples/qwen3_reranker_transformers.py:82
↓ 1 callersMethodencode
( self, sentences: Sequence[str], *, task_name: str, prompt_type: Prom
evaluation/qwen3_embedding_model.py:248
↓ 1 callersMethodformat_instruction
(self, instruction, query, doc)
examples/qwen3_reranker_vllm.py:52
↓ 1 callersMethodformat_instruction
(self, instruction, query, doc)
examples/qwen3_reranker_transformers.py:53
↓ 1 callersMethodformat_instruction
(self, instruction, prompt_type)
evaluation/qwen3_embedding_model.py:242
↓ 1 callersMethodformat_instruction
(self, instruction, query, doc)
evaluation/qwen3_reranker_model.py:54
↓ 1 callersMethodforward
( self, input_ids: torch.LongTensor, attention_mask: torch.Tensor, **kwargs
evaluation/qwen3_embedding_model.py:74
↓ 1 callersMethodget_detailed_instruct
(self, task_description: str, query: str)
examples/qwen3_embedding_transformers.py:38
↓ 1 callersMethodget_detailed_instruct
(self, task_description: str, query: str)
examples/qwen3_embedding_vllm.py:15
↓ 1 callersMethodget_instruction
(self, task_name, prompt_type)
evaluation/qwen3_embedding_model.py:222
↓ 1 callersFunctionget_model
(model_path: str, model_name: str, precision: str = 'fp16', **kwargs)
evaluation/run_mteb.py:250
↓ 1 callersFunctionget_model
(model_name: str, precision: str = 'fp16', **kwargs)
evaluation/run_mteb_reranking.py:103
↓ 1 callersFunctionget_tasks
(names: list[str] | None, languages: list[str] | None = None, benchmark: str | None = None)
evaluation/summary.py:14
↓ 1 callersFunctionget_tasks
(names: list[str] | None, languages: list[str] | None = None, benchmark: str | None = None)
evaluation/run_mteb.py:242
↓ 1 callersFunctionget_tasks
(names: list[str] | None, languages: list[str] | None = None, benchmark: str | None = None)
evaluation/run_mteb_reranking.py:87
↓ 1 callersMethodlast_token_pool
(self, last_hidden_states: Tensor, attention_mask: Tensor)
examples/qwen3_embedding_transformers.py:28
↓ 1 callersFunctionmain
()
evaluation/run_mteb.py:340
↓ 1 callersFunctionmain
()
evaluation/run_mteb_reranking.py:196
↓ 1 callersMethodprocess_batch
(self, pairs, **kwargs)
evaluation/qwen3_reranker_model.py:64
↓ 1 callersMethodprocess_inputs
(self, pairs)
examples/qwen3_reranker_transformers.py:59
↓ 1 callersFunctionrun_eval
(model, tasks: list, args: EvalArguments, **kwargs)
evaluation/run_mteb.py:288
↓ 1 callersFunctionrun_eval
(model, tasks: list, args: EvalArguments, **kwargs)
evaluation/run_mteb_reranking.py:109
↓ 1 callersMethodstop
(self)
examples/qwen3_reranker_vllm.py:89
↓ 1 callersMethodstop
(self)
examples/qwen3_embedding_vllm.py:33
↓ 1 callersMethodtokenize
(self, texts, max_length: int, prompt=None)
evaluation/qwen3_embedding_model.py:68
Method__init__
(self, model_name_or_path, instruction=None, use_fp16: bool = True, use_cuda: bool = True, max_length=8192)
examples/qwen3_embedding_transformers.py:15
Method__init__
(self, model_name_or_path, instruction="Given the user query, retrieval the relevant passages", **kwargs)
examples/qwen3_reranker_vllm.py:31
Method__init__
(self, model_name_or_path, instruction=None, max_length=8192)
examples/qwen3_embedding_vllm.py:9
Method__init__
( self, model_name_or_path: str, max_length: int = 4096, instruction=None,
examples/qwen3_reranker_transformers.py:30
Method__init__
( self, model: str, pooler_type: str = 'last', do_norm: bool = False,
evaluation/qwen3_embedding_model.py:25
Method__init__
(self, model_name, instruction="Given the user query, retrieval the relevant passages", **kwargs)
evaluation/qwen3_reranker_model.py:31
Method__post_init__
(self)
evaluation/run_mteb.py:228
Method__post_init__
(self)
evaluation/run_mteb_reranking.py:74
Function_encode_loop
( model: TransformersTextEmbedder, input_queue, output_queue, device: torch.device, qsize:
evaluation/qwen3_embedding_model.py:128
Function_evaluate_subset
( self, retriever, corpus, queries, relevant_docs, hf_subset: str, **kwargs )
evaluation/run_mteb.py:86
Method_generate_examples
Yields examples.
evaluation/mtebdata/RAR-b/TempReason-l2-fact/TempReason-l2-fact.py:61
Method_generate_examples
Yields examples.
evaluation/mtebdata/RAR-b/TempReason-l2-pure/TempReason-l2-pure.py:61
Method_generate_examples
Yields examples.
evaluation/mtebdata/RAR-b/TempReason-l2-context/TempReason-l2-context.py:61
Method_generate_examples
Yields examples.
evaluation/mtebdata/RAR-b/TempReason-l3-context/TempReason-l3-context.py:61
Method_generate_examples
Yields examples.
evaluation/mtebdata/RAR-b/winogrande/winogrande.py:61
Method_generate_examples
Yields examples.
evaluation/mtebdata/RAR-b/TempReason-l3-fact/TempReason-l3-fact.py:61
Method_generate_examples
Yields examples.
evaluation/mtebdata/RAR-b/TempReason-l3-pure/TempReason-l3-pure.py:61
Method_info
(self)
evaluation/mtebdata/RAR-b/TempReason-l2-fact/TempReason-l2-fact.py:25
Method_info
(self)
evaluation/mtebdata/RAR-b/TempReason-l2-pure/TempReason-l2-pure.py:25
Method_info
(self)
evaluation/mtebdata/RAR-b/TempReason-l2-context/TempReason-l2-context.py:25
Method_info
(self)
evaluation/mtebdata/RAR-b/TempReason-l3-context/TempReason-l3-context.py:25
Method_info
(self)
evaluation/mtebdata/RAR-b/winogrande/winogrande.py:25
Method_info
(self)
evaluation/mtebdata/RAR-b/TempReason-l3-fact/TempReason-l3-fact.py:25
Method_info
(self)
evaluation/mtebdata/RAR-b/TempReason-l3-pure/TempReason-l3-pure.py:25
Method_pooling_first
(hidden_state: torch.Tensor, _)
evaluation/qwen3_embedding_model.py:105
Method_pooling_last
(hidden_state: torch.Tensor, attention_mask: torch.Tensor)
evaluation/qwen3_embedding_model.py:94
Method_pooling_last_left
(hidden_state: torch.Tensor, _)
evaluation/qwen3_embedding_model.py:109
Method_pooling_last_right
(hidden_state: torch.Tensor, attention_mask: torch.Tensor)
evaluation/qwen3_embedding_model.py:113
Method_pooling_mean
(hidden_state: torch.Tensor, attention_mask: torch.Tensor)
evaluation/qwen3_embedding_model.py:120
Method_receive
(oq, timeout=0.00125)
evaluation/qwen3_embedding_model.py:269
Method_split_generators
Returns SplitGenerators.
evaluation/mtebdata/RAR-b/TempReason-l2-fact/TempReason-l2-fact.py:40
Method_split_generators
Returns SplitGenerators.
evaluation/mtebdata/RAR-b/TempReason-l2-pure/TempReason-l2-pure.py:40
Method_split_generators
Returns SplitGenerators.
evaluation/mtebdata/RAR-b/TempReason-l2-context/TempReason-l2-context.py:40
Method_split_generators
Returns SplitGenerators.
evaluation/mtebdata/RAR-b/TempReason-l3-context/TempReason-l3-context.py:40
Method_split_generators
Returns SplitGenerators.
evaluation/mtebdata/RAR-b/winogrande/winogrande.py:40
Method_split_generators
Returns SplitGenerators.
evaluation/mtebdata/RAR-b/TempReason-l3-fact/TempReason-l3-fact.py:40
Method_split_generators
Returns SplitGenerators.
evaluation/mtebdata/RAR-b/TempReason-l3-pure/TempReason-l3-pure.py:40
Functionevaluate
( self, model, split: str = "test", subsets_to_run: list | None = None, *, encode_kwar
evaluation/run_mteb.py:48
Methodpredict
( self, sentences: list[tuple[str, str]] | list[list[str]], batch_size: int = None,
evaluation/qwen3_reranker_model.py:94
Methodstart
(self)
evaluation/qwen3_reranker_model.py:91
Methodstop
(self)
evaluation/qwen3_reranker_model.py:109