[examples/deepspeed] fix renamed api (#21283)

2023-01-24 09:54:33 -08:00
parent e2e393c6f2
commit 9286039c2a
1 changed files with 1 additions and 1 deletions
--- a/tests/deepspeed/test_model_zoo.py
+++ b/tests/deepspeed/test_model_zoo.py
@@ -267,7 +267,7 @@ def make_task_cmds():
            --dataset_name hf-internal-testing/cats_vs_dogs_sample
            --remove_unused_columns False
            --max_steps 10
-            --feature_extractor_name {DS_TESTS_DIRECTORY}/vit_feature_extractor.json
+            --image_processor_name {DS_TESTS_DIRECTORY}/vit_feature_extractor.json
        """,
    )