Skip to content

Commit a011954

Browse files
committed
Revert "Disable ONNX tests (#2509)"
This reverts commit a0549fe.
1 parent 8f21a9f commit a011954

File tree

3 files changed

+15
-0
lines changed

3 files changed

+15
-0
lines changed

.github/workflows/pr_tests.yml

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,11 @@ jobs:
3131
runner: docker-cpu
3232
image: diffusers/diffusers-flax-cpu
3333
report: flax_cpu
34+
- name: Fast ONNXRuntime CPU tests on Ubuntu
35+
framework: onnxruntime
36+
runner: docker-cpu
37+
image: diffusers/diffusers-onnxruntime-cpu
38+
report: onnx_cpu
3439
- name: PyTorch Example CPU tests on Ubuntu
3540
framework: pytorch_examples
3641
runner: docker-cpu

.github/workflows/push_tests.yml

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,11 @@ jobs:
2929
runner: docker-tpu
3030
image: diffusers/diffusers-flax-tpu
3131
report: flax_tpu
32+
- name: Slow ONNXRuntime CUDA tests on Ubuntu
33+
framework: onnxruntime
34+
runner: docker-gpu
35+
image: diffusers/diffusers-onnxruntime-cuda
36+
report: onnx_cuda
3237

3338
name: ${{ matrix.config.name }}
3439

.github/workflows/push_tests_fast.yml

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,11 @@ jobs:
2929
runner: docker-cpu
3030
image: diffusers/diffusers-flax-cpu
3131
report: flax_cpu
32+
- name: Fast ONNXRuntime CPU tests on Ubuntu
33+
framework: onnxruntime
34+
runner: docker-cpu
35+
image: diffusers/diffusers-onnxruntime-cpu
36+
report: onnx_cpu
3237
- name: PyTorch Example CPU tests on Ubuntu
3338
framework: pytorch_examples
3439
runner: docker-cpu

0 commit comments

Comments
 (0)