- api端口参数说明： src/api
bash
-h, --help 显示帮助信息并退出
--model_name_or_path MODEL_NAME_OR_PATH
模型权重的路径或标识符，来自 huggingface.co/models 或 modelscope.cn/models。 (默认: None)
--adapter_name_or_path ADAPTER_NAME_OR_PATH
适配器权重的路径或标识符，来自 huggingface.co/models。 (默认: None)
--cache_dir CACHE_DIR
存储从 huggingface.co 或 modelscope.cn 下载的预训练模型的位置。 (默认: None)
--use_fast_tokenizer [USE_FAST_TOKENIZER]
是否使用基于 tokenizers 库支持的快速分词器。 (默认: True)
--no_use_fast_tokenizer
是否使用基于 tokenizers 库支持的快速分词器。 (默认: False)
--split_special_tokens [SPLIT_SPECIAL_TOKENS]
是否在标记化过程中拆分特殊标记。 (默认: False)
--model_revision MODEL_REVISION
要使用的特定模型版本（可以是分支名称、标签名称或提交 ID）。 (默认: main)
--quantization_bit QUANTIZATION_BIT
量化模型所需的位数。 (默认: None)
--quantization_type {fp4,nf4}
在 int4 训练中使用的量化数据类型。 (默认: nf4)
--double_quantization [DOUBLE_QUANTIZATION]
是否在 int4 训练中使用双重量化。 (默认: True)
--no_double_quantization
是否在 int4 训练中使用双重量化。 (默认: False)
--rope_scaling {linear,dynamic}
采用缩放的旋转位置嵌入。 (默认: None)
--flash_attn [FLASH_ATTN]
启用 FlashAttention-2 以加速训练。 (默认: False)
--shift_attn [SHIFT_ATTN]
启用由 LongLoRA 提出的 Shift Short Attention（S^2-Attn）。 (默认: False)
--hf_hub_token HF_HUB_TOKEN
用于登录 Hugging Face Hub 的身份验证令牌。 (默认: None)
--ms_hub_token MS_HUB_TOKEN
用于登录 ModelScope Hub 的身份验证令牌。 (默认: None)
--template TEMPLATE 用于构建训练和推断提示的模板。 (默认: None)
--dataset DATASET 要使用的提供的数据集名称。使用逗号分隔多个数据集。 (默认: None)
--dataset_dir DATASET_DIR
包含数据集的文件夹路径。 (默认: data)
--split SPLIT 用于训练和评估的数据集拆分。 (默认: train)
--cutoff_len CUTOFF_LEN
标记化后模型输入的最大长度。 (默认: 1024)
--reserved_label_len RESERVED_LABEL_LEN
标记化后为标签保留的最大长度。 (默认: 1)
--train_on_prompt [TRAIN_ON_PROMPT]
是否在提示上禁用掩码。 (默认: False)
--streaming [STREAMING]
启用数据集流式处理。 (默认: False)
--buffer_size BUFFER_SIZE
在数据集流式处理中用于随机抽样的缓冲区大小。 (默认: 16384)
--mix_strategy {concat,interleave_under,interleave_over}
数据集混合策略（连接/交错）（欠采样/过采样）。 (默认: concat)
--interleave_probs INTERLEAVE_PROBS
从数据集中抽样数据的概率。使用逗号分隔多个数据集。 (默认: None)
--overwrite_cache [OVERWRITE_CACHE]
覆盖缓存的训练和评估集。 (默认: False)
--preprocessing_num_workers PREPROCESSING_NUM_WORKERS
用于预处理的进程数。 (默认: None)
--max_samples MAX_SAMPLES
用于调试目的，截断每个数据集的示例数。 (默认: None)
--eval_num_beams EVAL_NUM_BEAMS
用于评估的束搜索数。该参数将传递给 `model.generate`。 (默认: None)
--ignore_pad_token_for_loss [IGNORE_PAD_TOKEN_FOR_LOSS]
是否在损失计算中忽略与填充标签相对应的标记。 (默认: True)
--no_ignore_pad_token_for_loss
是否在损失计算中忽略与填充标签相对应的标记。 (默认: False)
--val_size VAL_SIZE 开发集的大小，应为整数或范围为 `[0,1)` 的浮点数。 (默认: 0)
--sft_packing [SFT_PACKING]
在监督微调阶段对问题和答案进行打包。 (默认: False)
--cache_path CACHE_PATH
保存或加载预处理数据集的路径。 (默认: None)
--export_dir EXPORT_DIR
保存导出模型的目录路径。 (默认: None)
--export_size EXPORT_SIZE
导出模型的文件分片大小（以 GB 为单位）。 (默认: 1)
--export_quantization_bit EXPORT_QUANTIZATION_BIT
用于量化导出模型的位数。 (默认: None)
--export_quantization_dataset EXPORT_QUANTIZATION_DATASET
用于量化导出模型的数据集路径或数据集名称。 (默认: None)
--export_quantization_nsamples EXPORT_QUANTIZATION_NSAMPLES
用于量化的样本数。 (默认: 128)
--export_quantization_maxlen EXPORT_QUANTIZATION_MAXLEN
用于量化的模型输入的最大长度。 (默认: 1024)
--dpo_beta DPO_BETA DPO 损失的 beta 参数。 (默认: 0.1)
--dpo_loss {sigmoid,hinge}
要使用的 DPO 损失类型。 (默认: sigmoid)
--dpo_ftx DPO_FTX DPO 训练中的监督微调损失系数。 (默认: 0)
--ppo_buffer_size PPO_BUFFER_SIZE
在 PPO 优化步骤中创建经验缓冲区的小批次数。 (默认: 1)
--ppo_epochs PPO_EPOCHS
在 PPO 优化步骤中执行的周期数。 (默认: 4)
--ppo_logger PPO_LOGGER
在 PPO 训练中记录日志，使用 "wandb" 或 "tensorboard"。 (默认: None)
--ppo_score_norm [PPO_SCORE_NORM]
在 PPO 训练中使用分数归一化。 (默认: False)
--ppo_target PPO_TARGET
用于自适应 KL 控制的目标 KL 值，在 PPO 训练中。 (默认: 6.0)
--ppo_whiten_rewards [PPO_WHITEN_REWARDS]
在 PPO 训练中计算优势之前是否漂白奖励。 (默认: False)
--ref_model REF_MODEL
用于 PPO 或 DPO 训练的参考模型路径。 (默认: None)
--ref_model_adapters REF_MODEL_ADAPTERS
参考模型的适配器路径。 (默认: None)
--ref_model_quantization_bit REF_MODEL_QUANTIZATION_BIT
用于量化参考模型的位数。 (默认: None)
--reward_model REWARD_MODEL
用于 PPO 训练的奖励模型路径。 (默认: None)
--reward_model_adapters REWARD_MODEL_ADAPTERS
奖励模型的适配器路径。 (默认: None)
--reward_model_quantization_bit REWARD_MODEL_QUANTIZATION_BIT
用于量化奖励模型的位数。 (默认: None)
--reward_model_type {lora,full,api}
在 PPO 训练中使用的奖励模型类型。LoRA 模型仅支持 LoRA 训练。 (默认: lora)
--additional_target ADDITIONAL_TARGET
除 LoRA 层之外要设置为可训练并保存在最终检查点中的模块的名称。 (默认: None)
--lora_alpha LORA_ALPHA
LoRA 微调的比例因子（默认为 lora_rank * 2）。 (默认: None)
--lora_dropout LORA_DROPOUT
LoRA 微调的 dropout 率。 (默认: 0.1)
--lora_rank LORA_RANK
LoRA 微调的内在维度。 (默认: 8)
--lora_target LORA_TARGET
要应用 LoRA 的目标模块的名称。使用逗号分隔多个模块。 (默认: None)
--create_new_adapter [CREATE_NEW_ADAPTER]
是否创建具有随机初始化权重的新适配器。 (默认: False)
--name_module_trainable NAME_MODULE_TRAINABLE
部分参数（冻结）微调的可训练模块的名称。使用逗号分隔多个模块。 (默认: mlp)
--num_layer_trainable NUM_LAYER_TRAINABLE
部分参数（冻结）微调的可训练层数。 (默认: 3)
--stage {pt,sft,rm,ppo,dpo}
训练中将执行哪个阶段。 (默认: sft)
--finetuning_type {lora,freeze,full}
使用哪种微调方法。 (默认: lora)
--upcast_layernorm [UPCAST_LAYERNORM]
是否在 fp32 中上升 layernorm 权重。 (默认: False)
--plot_loss [PLOT_LOSS]
是否在微调后绘制训练损失。 (默认: False)
--do_sample [DO_SAMPLE]
是否使用抽样，否则使用贪婪解码。 (默认: True)
--no_do_sample 是否使用抽样，否则使用贪婪解码。 (默认: False)
--temperature TEMPERATURE
用于调整下一个标记概率的值。 (默认: 0.95)
--top_p TOP_P 保留的最概然令牌集，其概率加起来达到 top_p 或更高。 (默认: 0.7)
--top_k TOP_K 保留的最高概率词汇令牌数量，用于 top-k 过滤。 (默认: 50)
--num_beams NUM_BEAMS
用于波束搜索的波束数量。1 表示不进行波束搜索。 (默认: 1)
--max_length MAX_LENGTH
生成令牌的最大长度。可以被 max_new_tokens 覆盖。 (默认: 512)
--max_new_tokens MAX_NEW_TOKENS
生成的令牌的最大数量，忽略提示中的令牌数量。 (默认: 512)
--repetition_penalty REPETITION_PENALTY
重复惩罚的参数。1.0 表示无惩罚。 (默认: 1.0)
--length_penalty LENGTH_PENALTY
用于基于波束的生成的长度的指数惩罚。 (默认: 1.0)