Spaces:
Runtime error
Runtime error
blackwingedkite
commited on
Commit
•
a785f6e
1
Parent(s):
ebcadc0
Update app.py
Browse files
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
|
8 |
--do_train \
|
9 |
--dataset mydata \
|
10 |
-
--template
|
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 \
|