add export_device in webui #3333

2024-04-25 19:02:32 +08:00
1 parent a3aad4b4f0
commit 3a7c1286ce
@@ -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