Spaces:
Runtime error
Runtime error
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 \
|