File tree 4 files changed +20
-3780
lines changed
4 files changed +20
-3780
lines changed Original file line number Diff line number Diff line change @@ -10,7 +10,7 @@ torchrun --standalone --nnodes=1 --nproc-per-node=$NUM_GPUS main.py \
10
10
--overwrite_cache \
11
11
--prompt_column content \
12
12
--response_column summary \
13
- --model_name_or_path chatglm2-6b \
13
+ --model_name_or_path THUDM/ chatglm2-6b \
14
14
--ptuning_checkpoint ./output/$CHECKPOINT /checkpoint-$STEP \
15
15
--output_dir ./output/$CHECKPOINT \
16
16
--overwrite_output_dir \
Original file line number Diff line number Diff line change @@ -178,7 +178,7 @@ def preprocess_function_eval(examples):
178
178
return model_inputs
179
179
180
180
def preprocess_function_train (examples ):
181
- max_seq_length = data_args .max_source_length + data_args .max_target_length
181
+ max_seq_length = data_args .max_source_length + data_args .max_target_length + 1
182
182
183
183
model_inputs = {
184
184
"input_ids" : [],
@@ -335,7 +335,7 @@ def compute_metrics(eval_preds):
335
335
tokenizer = tokenizer ,
336
336
data_collator = data_collator ,
337
337
compute_metrics = compute_metrics if training_args .predict_with_generate else None ,
338
- save_prefixencoder = model_args .pre_seq_len is not None
338
+ save_changed = model_args .pre_seq_len is not None
339
339
)
340
340
341
341
# Training
You can’t perform that action at this time.
0 commit comments