SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Stas Bekman
2023-01-24 09:54:33 -08:00
committed by GitHub
parent e2e393c6f2
commit 9286039c2a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/deepspeed/test_model_zoo.py
View File

@@ -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
""", """,
) )