add export_device in webui #3333

Former-commit-id: 3a7c1286ce
This commit is contained in:
hiyouga
2024-04-25 19:02:32 +08:00
parent 0158ab3c27
commit 84031c5bf9
7 changed files with 57 additions and 22 deletions

View File

@@ -8,4 +8,5 @@ CUDA_VISIBLE_DEVICES=0 python ../../src/export_model.py \
--finetuning_type lora \
--export_dir ../../models/llama2-7b-sft \
--export_size 2 \
--export_device cpu \
--export_legacy_format False