Views
No views yet
1NPROC_PER_NODE=8 \
2CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 \
3MASTER_PORT=29500 \
4swift sft \
5 --model_type llama3-8b-instruct \
6 --learning_rate 2e-5 \
7 --sft_type lora \
8 --dataset msagent-pro \
9 --gradient_checkpointing true \
10 --gradient_accumulation_steps 8 \
11 --deepspeed default-zero3 \
12 --lora_target_modules ALL \
13 --use_loss_scale true \
14 --save_strategy epoch \
15 --batch_size 1 \
16 --num_train_epochs 2 \
17 --max_length 4096 \
18 --preprocess_num_proc 4 \
19 --use_loss_scale true \
20 --loss_scale_config_path agent-flan \
21 --ddp_backend nccl \| Model | ToolBench (in-domain) | ToolBench (out-of-domain) | |||||||
|---|---|---|---|---|---|---|---|---|---|
| Plan.EM | Act.EM | HalluRate (lower is better) | Avg.F1 | R-L | Plan.EM | Act.EM | HalluRate (lower is better) | Avg.F1 | |
| llama3-8b-instruct | 74.22 | 36.17 | 15.68 | 20.0 | 12.14 | 69.47 | 34.21 | 14.72 | 20.25 |
| llama3-8b-agent-instruct-v2 | 85.15 | 58.1 | 1.57 | 52.10 | 26.02 | 85.79 | 59.43 | 2.56 | 52.19 |
1USE_HF=True swift deploy \
2 --model_id_or_path modelscope/llama3-8b-agent-instruct-v2 \
3 --model_type llama3-8b-instruct \
4 --infer_backend vllm \
5 --tools_prompt toolbench