Fix failing DeepSpeed model zoo tests (#30112)
* fix sequence length errors * fix label column name error for vit * fix the lm_head embedding!=linear layer mismatches for Seq2Seq models
This commit is contained in:
committed by
GitHub
parent
2f12e40822
commit
4e3490f79b
@@ -236,6 +236,8 @@ def make_task_cmds():
|
||||
--train_file {data_dir_wmt}/train.json
|
||||
--source_lang en
|
||||
--target_lang ro
|
||||
--max_source_length 12
|
||||
--max_target_length 12
|
||||
""",
|
||||
"sum": f"""
|
||||
{scripts_dir}/summarization/run_summarization.py
|
||||
@@ -269,6 +271,7 @@ def make_task_cmds():
|
||||
--remove_unused_columns False
|
||||
--max_steps 10
|
||||
--image_processor_name {DS_TESTS_DIRECTORY}/vit_feature_extractor.json
|
||||
--label_column_name labels
|
||||
""",
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user