blackwingedkite commited on
Commit
6725046
1 Parent(s): fd30a7e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,11 +1,11 @@
1
  !CUDA_VISIBLE_DEVICES=0 python src/train_bash.py \
2
  --stage sft \
3
- --model_name_or_path ziqingyang/chinese-alpaca-2-7b \
4
  --do_train \
5
  --dataset mydata \
6
- --template default \
7
  --finetuning_type lora \
8
- --lora_target q_proj,v_proj \
9
  --output_dir ckpoint \
10
  --overwrite_cache \
11
  --auto_find_batch_size=False \
 
1
  !CUDA_VISIBLE_DEVICES=0 python src/train_bash.py \
2
  --stage sft \
3
+ --model_name_or_path THUDM/chatglm2-6b \ #ziqingyang/chinese-alpaca-2-7b
4
  --do_train \
5
  --dataset mydata \
6
+ --template chatglm2 \
7
  --finetuning_type lora \
8
+ --lora_target query_key_value \
9
  --output_dir ckpoint \
10
  --overwrite_cache \
11
  --auto_find_batch_size=False \