Arguments

class FlagEmbedding.finetune.reranker.decoder_only.layerwise.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 = 'from_raw_model', 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, start_layer: int = 8, head_multi: bool = False, head_type: str = 'simple')

Model argument class for decoder only reranker.