Skip to content

Commit e862480

Browse files
committed
update
1 parent 0cee362 commit e862480

2 files changed

Lines changed: 17 additions & 17 deletions

File tree

.github/workflows/pr_tests.yml

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,8 @@ concurrency:
2323
env:
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

3030
jobs:
@@ -79,7 +79,7 @@ jobs:
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

.github/workflows/pr_tests_gpu.yml

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -1,19 +1,19 @@
11
name: Fast GPU Tests on PR
22

33
on:
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

1919
concurrency:

0 commit comments

Comments
 (0)