11name : Fast GPU Tests on main
22
33on :
4+ pull_request :
5+ branches : main
6+ paths :
7+ - " src/diffusers/models/modeling_utils.py"
8+ - " src/diffusers/models/model_loading_utils.py"
9+ - " src/diffusers/pipelines/pipeline_utils.py"
10+ - " src/diffusers/pipeline_loading_utils.py"
411 workflow_dispatch :
512 push :
613 branches :
@@ -160,6 +167,7 @@ jobs:
160167 path : reports
161168
162169 flax_tpu_tests :
170+ if : ${{ github.event_name != 'pull_request' }}
163171 name : Flax TPU Tests
164172 runs-on :
165173 group : gcp-ct5lp-hightpu-8t
@@ -208,6 +216,7 @@ jobs:
208216 path : reports
209217
210218 onnx_cuda_tests :
219+ if : ${{ github.event_name != 'pull_request' }}
211220 name : ONNX CUDA Tests
212221 runs-on :
213222 group : aws-g4dn-2xlarge
@@ -256,6 +265,7 @@ jobs:
256265 path : reports
257266
258267 run_torch_compile_tests :
268+ if : ${{ github.event_name != 'pull_request' }}
259269 name : PyTorch Compile CUDA tests
260270
261271 runs-on :
@@ -299,6 +309,7 @@ jobs:
299309 path : reports
300310
301311 run_xformers_tests :
312+ if : ${{ github.event_name != 'pull_request' }}
302313 name : PyTorch xformers CUDA tests
303314
304315 runs-on :
@@ -349,7 +360,6 @@ jobs:
349360 container :
350361 image : diffusers/diffusers-pytorch-cuda
351362 options : --gpus 0 --shm-size "16gb" --ipc host
352-
353363 steps :
354364 - name : Checkout diffusers
355365 uses : actions/checkout@v3
@@ -359,7 +369,6 @@ jobs:
359369 - name : NVIDIA-SMI
360370 run : |
361371 nvidia-smi
362-
363372 - name : Install dependencies
364373 run : |
365374 python -m venv /opt/venv && export PATH="/opt/venv/bin:$PATH"
0 commit comments