From c083208a6e1d567c99d039f844fbc98d2fe2eea7 Mon Sep 17 00:00:00 2001 From: Cemberk Date: Wed, 11 Sep 2024 09:39:38 -0500 Subject: [PATCH] Update fork-maintenance.yml --- .github/workflows/fork-maintenance.yml | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/.github/workflows/fork-maintenance.yml b/.github/workflows/fork-maintenance.yml index 3c32f59cb27..09f5385b455 100644 --- a/.github/workflows/fork-maintenance.yml +++ b/.github/workflows/fork-maintenance.yml @@ -36,8 +36,7 @@ jobs: pr_branch_prefix: "scheduled-merge" requirements_command: > sudo sed -i 's/torchaudio//g' examples/pytorch/_tests_requirements.txt && pip install -r examples/pytorch/_tests_requirements.txt && pip install --no-cache-dir GPUtil azureml azureml-core tokenizers ninja cerberus sympy sacremoses sacrebleu==1.5.1 sentencepiece scipy scikit-learn urllib3 && pip install huggingface_hub datasets && pip install parameterized && pip install -e . - unit_test_command: > - pytest + unit_test_command: pytest performance_test_command: > echo "python examples/pytorch/language-modeling/run_mlm.py --model_name_or_path bert-base-uncased --dataset_name wikitext --dataset_config_name wikitext-2-raw-v1 --do_train --do_eval --output_dir /tmp/test-mlm --per_device_train_batch_size 8 --per_device_eval_batch_size 8 --max_steps 500" docker_image: rocm/pytorch:latest