[MNT] Fix infer scripts

This commit is contained in:
songpx
2023-06-06 21:00:30 +08:00
parent 94b70a363a
commit 6bd5919af1
2 changed files with 72 additions and 47 deletions

View File

@ -3,5 +3,5 @@ python infer.py \
--load_8bit True \
--base_model 'minlik/chinese-llama-7b-merged' \
--lora_weights 'entity303/lawgpt-lora-7b' \
--infer_data_path './resources/example_infer_data.json' \
--prompt_template 'law_template'
--prompt_template 'law_template' \
--infer_data_path './resources/example_infer_data.json'