Compare commits

...

3 Commits

Author SHA1 Message Date
Adrien
99dd8579f7 fixup
Signed-off-by: Adrien <adrien@huggingface.co>
2024-07-31 14:09:02 +02:00
Adrien
1972b80b7a Merge branch 'main' into test-cache
Signed-off-by: Adrien <adrien@huggingface.co>
2024-07-31 14:05:13 +02:00
Adrien
9e638d8903 test ci cache
Signed-off-by: Adrien <adrien@huggingface.co>
2024-07-31 11:44:17 +02:00
9 changed files with 18 additions and 18 deletions

View File

@@ -21,7 +21,7 @@ env:
jobs:
test-build-docker-images:
runs-on:
group: aws-general-8-plus
group: aws-general-8-plus-cache
if: github.event_name == 'pull_request'
steps:
- name: Set up Docker Buildx
@@ -52,7 +52,7 @@ jobs:
build-and-push-docker-images:
runs-on:
group: aws-general-8-plus
group: aws-general-8-plus-cache
if: github.event_name != 'pull_request'
permissions:

View File

@@ -20,7 +20,7 @@ jobs:
setup_torch_cuda_pipeline_matrix:
name: Setup Torch Pipelines CUDA Slow Tests Matrix
runs-on:
group: aws-general-8-plus
group: aws-general-8-plus-cache
container:
image: diffusers/diffusers-pytorch-cpu
outputs:

View File

@@ -16,7 +16,7 @@ jobs:
setup_pr_tests:
name: Setup PR Tests
runs-on:
group: aws-general-8-plus
group: aws-general-8-plus-cache
container:
image: diffusers/diffusers-pytorch-cpu
options: --shm-size "16gb" --ipc host -v /mnt/hf_cache:/mnt/cache/
@@ -75,7 +75,7 @@ jobs:
matrix:
modules: ${{ fromJson(needs.setup_pr_tests.outputs.matrix) }}
runs-on:
group: aws-general-8-plus
group: aws-general-8-plus-cache
container:
image: diffusers/diffusers-pytorch-cpu
options: --shm-size "16gb" --ipc host -v /mnt/hf_cache:/mnt/cache/
@@ -125,7 +125,7 @@ jobs:
config:
- name: Hub tests for models, schedulers, and pipelines
framework: hub_tests_pytorch
runner: aws-general-8-plus
runner: aws-general-8-plus-cache
image: diffusers/diffusers-pytorch-cpu
report: torch_hub

View File

@@ -72,7 +72,7 @@ jobs:
name: LoRA - ${{ matrix.lib-versions }}
runs-on:
group: aws-general-8-plus
group: aws-general-8-plus-cache
container:
image: diffusers/diffusers-pytorch-cpu

View File

@@ -82,17 +82,17 @@ jobs:
report: torch_cpu_pipelines
- name: Fast PyTorch Models & Schedulers CPU tests
framework: pytorch_models
runner: aws-general-8-plus
runner: aws-general-8-plus-cache
image: diffusers/diffusers-pytorch-cpu
report: torch_cpu_models_schedulers
- name: Fast Flax CPU tests
framework: flax
runner: aws-general-8-plus
runner: aws-general-8-plus-cache
image: diffusers/diffusers-flax-cpu
report: flax_cpu
- name: PyTorch Example CPU tests
framework: pytorch_examples
runner: aws-general-8-plus
runner: aws-general-8-plus-cache
image: diffusers/diffusers-pytorch-cpu
report: torch_example_cpu
@@ -182,7 +182,7 @@ jobs:
- name: Hub tests for models, schedulers, and pipelines
framework: hub_tests_pytorch
runner:
group: aws-general-8-plus
group: aws-general-8-plus-cache
image: diffusers/diffusers-pytorch-cpu
report: torch_hub

View File

@@ -20,7 +20,7 @@ jobs:
setup_torch_cuda_pipeline_matrix:
name: Setup Torch Pipelines CUDA Slow Tests Matrix
runs-on:
group: aws-general-8-plus
group: aws-general-8-plus-cache
container:
image: diffusers/diffusers-pytorch-cpu
outputs:

View File

@@ -29,22 +29,22 @@ jobs:
config:
- name: Fast PyTorch CPU tests on Ubuntu
framework: pytorch
runner: aws-general-8-plus
runner: aws-general-8-plus-cache
image: diffusers/diffusers-pytorch-cpu
report: torch_cpu
- name: Fast Flax CPU tests on Ubuntu
framework: flax
runner: aws-general-8-plus
runner: aws-general-8-plus-cache
image: diffusers/diffusers-flax-cpu
report: flax_cpu
- name: Fast ONNXRuntime CPU tests on Ubuntu
framework: onnxruntime
runner: aws-general-8-plus
runner: aws-general-8-plus-cache
image: diffusers/diffusers-onnxruntime-cpu
report: onnx_cpu
- name: PyTorch Example CPU tests on Ubuntu
framework: pytorch_examples
runner: aws-general-8-plus
runner: aws-general-8-plus-cache
image: diffusers/diffusers-pytorch-cpu
report: torch_example_cpu

View File

@@ -101,7 +101,7 @@ _deps = [
"filelock",
"flax>=0.4.1",
"hf-doc-builder>=0.3.0",
"huggingface-hub>=0.23.2",
"huggingface-hub>=0.24.5",
"requests-mock==1.10.0",
"importlib_metadata",
"invisible-watermark>=0.2.0",

View File

@@ -9,7 +9,7 @@ deps = {
"filelock": "filelock",
"flax": "flax>=0.4.1",
"hf-doc-builder": "hf-doc-builder>=0.3.0",
"huggingface-hub": "huggingface-hub>=0.23.2",
"huggingface-hub": "huggingface-hub>=0.24.5",
"requests-mock": "requests-mock==1.10.0",
"importlib_metadata": "importlib_metadata",
"invisible-watermark": "invisible-watermark>=0.2.0",