Compare commits

...

13 Commits

Author SHA1 Message Date
ydshieh fae4ab256e fix 2024-04-16 16:13:54 +02:00
ydshieh 258a062c46 fix 2024-04-16 15:57:00 +02:00
ydshieh 3b27b04f17 check whisper 2024-04-16 15:05:50 +02:00
ydshieh 77893abf6c fix 2024-04-16 14:58:49 +02:00
ydshieh 63f77a1ff1 fix 2024-04-16 14:50:11 +02:00
sanchit-gandhi 5e04425a02 Merge remote-tracking branch 'origin/whisper-slow-tests' into whisper-slow-tests 2024-04-15 17:27:43 +01:00
sanchit-gandhi 5739f540f6 use ungated dataset 2024-04-15 17:27:38 +01:00
ydshieh 723aa337fd update expected values 2024-04-11 16:09:30 +02:00
sanchit-gandhi d29547a32c move logits to cpu 2024-04-09 23:02:52 +01:00
sanchit-gandhi 2ca8ecf0a7 move model to device 2024-04-09 22:56:30 +01:00
sanchit-gandhi 861d691206 more fixes 2024-04-09 22:47:59 +01:00
sanchit-gandhi fd1b7e2a6c style 2024-04-09 22:41:17 +01:00
sanchit-gandhi 6fc457976e fix tests 2024-04-09 22:37:23 +01:00
4 changed files with 144 additions and 169 deletions

View File

@ -80,7 +80,7 @@ jobs:
- name: Run all tests on GPU
working-directory: /transformers
run: python3 -m pytest -v --make-reports=${{ inputs.machine_type }}_tests_gpu_${{ matrix.folders }} tests/${{ matrix.folders }}
run: python3 -m pytest -v --make-reports=${{ inputs.machine_type }}_tests_gpu_${{ matrix.folders }} tests/models/whisper/test_modeling_whisper.py -k "WhisperModelIntegrationTests"
- name: Failure short reports
if: ${{ failure() }}

View File

@ -7,7 +7,7 @@ on:
- cron: "17 2 * * *"
push:
branches:
- run_scheduled_ci*
- check-whisper-slow-tests
jobs:
model-ci:
@ -17,43 +17,3 @@ jobs:
job: run_tests_gpu
slack_report_channel: "#transformers-ci-daily-models"
secrets: inherit
torch-pipeline:
name: Torch pipeline CI
uses: ./.github/workflows/self-scheduled.yml
with:
job: run_pipelines_torch_gpu
slack_report_channel: "#transformers-ci-daily-pipeline-torch"
secrets: inherit
tf-pipeline:
name: TF pipeline CI
uses: ./.github/workflows/self-scheduled.yml
with:
job: run_pipelines_tf_gpu
slack_report_channel: "#transformers-ci-daily-pipeline-tf"
secrets: inherit
example-ci:
name: Example CI
uses: ./.github/workflows/self-scheduled.yml
with:
job: run_examples_gpu
slack_report_channel: "#transformers-ci-daily-examples"
secrets: inherit
deepspeed-ci:
name: DeepSpeed CI
uses: ./.github/workflows/self-scheduled.yml
with:
job: run_all_tests_torch_cuda_extensions_gpu
slack_report_channel: "#transformers-ci-daily-deepspeed"
secrets: inherit
quantization-ci:
name: Quantization CI
uses: ./.github/workflows/self-scheduled.yml
with:
job: run_tests_quantization_torch_gpu
slack_report_channel: "#transformers-ci-daily-quantization"
secrets: inherit

File diff suppressed because one or more lines are too long

View File

@ -62,4 +62,5 @@ if __name__ == "__main__":
start = end
end = start + num_jobs_per_splits + (1 if idx < num_jobs % args.num_splits else 0)
model_splits.append(d[start:end])
model_splits = [["models/whisper"]]
print(model_splits)