Skip to content

GitLab

Unverified Commit 9286039c authored by Stas Bekman's avatar Stas Bekman Committed by GitHub
Browse files

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

parent e2e393c6
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/deepspeed/test_model_zoo.py
View file @ 9286039c
...@@ -267,7 +267,7 @@ def make_task_cmds(): ...@@ -267,7 +267,7 @@ def make_task_cmds():
--dataset_name hf-internal-testing/cats_vs_dogs_sample --dataset_name hf-internal-testing/cats_vs_dogs_sample
--remove_unused_columns False --remove_unused_columns False
--max_steps 10 --max_steps 10
--feature_extractor_name {DS_TESTS_DIRECTORY}/vit_feature_extractor.json --image_processor_name {DS_TESTS_DIRECTORY}/vit_feature_extractor.json
""", """,
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment