blackwingedkite commited on
Commit
a785f6e
1 Parent(s): ebcadc0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -6
app.py CHANGED
@@ -1,13 +1,9 @@
1
- checkpoint = "ziqingyang/chinese-alpaca-2-7b"
2
- tokenizer = AutoTokenizer.from_pretrained(checkpoint)
3
- model = AutoModelForMaskedLM.from_pretrained(checkpoint)
4
-
5
  !CUDA_VISIBLE_DEVICES=0 python src/train_bash.py \
6
  --stage sft \
7
- --model_name_or_path model \
8
  --do_train \
9
  --dataset mydata \
10
- --template chatglm2 \
11
  --finetuning_type lora \
12
  --lora_target q_proj,v_proj \
13
  --output_dir ckpoint \
 
 
 
 
 
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 \