Arguments#

class FlagEmbedding.finetune.reranker.decoder_only.base.RerankerModelArguments(model_name_or_path: str, config_name: str | None = None, tokenizer_name: str | None = None, cache_dir: str | None = None, trust_remote_code: bool = False, model_type: str = 'encoder', token: str = <factory>, use_lora: bool = True, lora_rank: int = 64, lora_alpha: float = 16, lora_dropout: float = 0.1, target_modules: ~typing.List[str] = <factory>, modules_to_save: ~typing.List[str] | None = None, use_flash_attn: bool = False, from_peft: str | None = None, raw_peft: ~typing.List[str] | None = None, save_merged_lora_model: bool = False)[source]#

Model argument class for decoder only reranker.