HIP_VISIBLE_DEVICES=7 \
python llm_infer.py \
    --ckpt_dir "/workspace/llm/Telechat/ms-swift/examples/pytorch/llm/output/checkpoint" \
    --load_dataset_config true \
    --max_length 2048 \
    --use_flash_attn false \
    --max_new_tokens 2048 \
    --temperature 0.5 \
    --top_p 0.7 \
    --repetition_penalty 1. \
    --do_sample true \
    --merge_lora false \
    --dtype fp16 \
    --stream false
