File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -23,8 +23,8 @@ concurrency:
2323env :
2424 DIFFUSERS_IS_CI : yes
2525 HF_XET_HIGH_PERFORMANCE : 1
26- OMP_NUM_THREADS : 1
27- MKL_NUM_THREADS : 1
26+ OMP_NUM_THREADS : 2
27+ MKL_NUM_THREADS : 2
2828 PYTEST_TIMEOUT : 60
2929
3030jobs :
7979 config :
8080 - name : Fast PyTorch Pipeline CPU tests
8181 framework : pytorch_pipelines
82- runner : aws-highmemory-32 -plus
82+ runner : aws-highmemory-64 -plus
8383 image : diffusers/diffusers-pytorch-cpu
8484 report : torch_cpu_pipelines
8585 - name : Fast PyTorch Models & Schedulers CPU tests
@@ -125,7 +125,7 @@ jobs:
125125 - name : Run fast PyTorch Pipeline CPU tests
126126 if : ${{ matrix.config.framework == 'pytorch_pipelines' }}
127127 run : |
128- pytest -n 16 --max-worker-restart=0 --dist=loadfile \
128+ pytest -n 24 --max-worker-restart=0 --dist=loadfile \
129129 -k "not Flax and not Onnx" \
130130 --make-reports=tests_${{ matrix.config.report }} \
131131 tests/pipelines
Original file line number Diff line number Diff line change 11name : Fast GPU Tests on PR
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"
11- - " src/diffusers/loaders/lora_base.py"
12- - " src/diffusers/loaders/lora_pipeline.py"
13- - " src/diffusers/loaders/peft.py"
14- - " tests/pipelines/test_pipelines_common.py"
15- - " tests/models/test_modeling_common.py"
16- - " examples/**/*.py"
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"
11+ # - "src/diffusers/loaders/lora_base.py"
12+ # - "src/diffusers/loaders/lora_pipeline.py"
13+ # - "src/diffusers/loaders/peft.py"
14+ # - "tests/pipelines/test_pipelines_common.py"
15+ # - "tests/models/test_modeling_common.py"
16+ # - "examples/**/*.py"
1717 workflow_dispatch :
1818
1919concurrency :
You can’t perform that action at this time.
0 commit comments