Spaces:
Runtime error
Runtime error
blackwingedkite
commited on
Commit
•
6725046
1
Parent(s):
fd30a7e
Update app.py
Browse files
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
|
7 |
--finetuning_type lora \
|
8 |
-
--lora_target
|
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 \
|