Skip to content

Commit

Permalink
Use stable-diffusion-xl-base-1.0 to validate graph persistence (#1169)
Browse files Browse the repository at this point in the history
<!-- This is an auto-generated comment: release notes by coderabbit.ai
-->
## Summary by CodeRabbit

- **Chores**
- Updated the model ID for the `run-examples` job in the workflow
configuration.
- Removed volume mapping for the `onediff-test` service in multiple
Docker Compose configuration files, affecting local data storage within
containers.
- Added a constraint for the installation of the `onediff-quant` package
to require `"numpy<2"`.
<!-- end of auto-generated comment: release notes by coderabbit.ai -->
  • Loading branch information
jackalcooper authored Dec 26, 2024
1 parent 6c7c122 commit 73eaeb7
Show file tree
Hide file tree
Showing 4 changed files with 3 additions and 10 deletions.
10 changes: 3 additions & 7 deletions .github/workflows/examples.yml
Original file line number Diff line number Diff line change
Expand Up @@ -178,7 +178,6 @@ jobs:
test-suite:
- diffusers_examples
- comfy
- webui
exclude:
- should-run-pro: false
image: onediff-pro:cu122
Expand Down Expand Up @@ -289,16 +288,13 @@ jobs:
- name: Install onediff-quant if needed
if: startsWith(matrix.image, 'onediff-pro')
run: |
docker exec ${{ env.CONTAINER_NAME }} python3 -m pip install --pre onediff-quant -f https://oneflow-pro.oss-cn-beijing.aliyuncs.com/onediff-quant/
docker exec ${{ env.CONTAINER_NAME }} python3 -m pip install --pre onediff-quant "numpy<2" -f https://oneflow-pro.oss-cn-beijing.aliyuncs.com/onediff-quant/
- name: Pip Install Requirements for ComfyUI & Test
if: matrix.test-suite == 'comfy'
run: |
docker exec ${{ env.CONTAINER_NAME }} python3 -m pip install -r tests/comfyui/requirements.txt --user
docker exec ${{ env.CONTAINER_NAME }} python3 -m pip install -r ComfyUI/requirements.txt --user
docker exec ${{ env.CONTAINER_NAME }} python3 -m pip uninstall -y transformer-engine
- name: Print pip package versions
run: |
docker exec ${{ env.CONTAINER_NAME }} python3 -m pip list
- name: Start ComfyUI Web Service
if: matrix.test-suite == 'comfy'
run: |
Expand Down Expand Up @@ -356,10 +352,10 @@ jobs:
- if: matrix.test-suite == 'diffusers_examples'
run: docker exec -w /src/onediff/onediff_diffusers_extensions ${{ env.CONTAINER_NAME }} python3 examples/text_to_image_controlnet.py --base=/share_nfs/hf_models/stable-diffusion-v1-5 --controlnet=/share_nfs/hf_models/sd-controlnet-canny --input_image=/share_nfs/hf_models/input_image_vermeer.png
- if: matrix.test-suite == 'diffusers_examples'
run: docker exec -w /src/onediff/onediff_diffusers_extensions ${{ env.CONTAINER_NAME }} bash examples/unet_save_and_load.sh --model_id=/share_nfs/hf_models/stable-diffusion-2-1
run: docker exec -w /src/onediff/onediff_diffusers_extensions ${{ env.CONTAINER_NAME }} bash examples/unet_save_and_load.sh --model_id=/share_nfs/hf_models/stable-diffusion-xl-base-1.0
- if: matrix.test-suite == 'diffusers_examples'
run: |
docker exec ${{ env.CONTAINER_NAME }} python3 -m pip install --user scikit-image
docker exec ${{ env.CONTAINER_NAME }} python3 -m pip install --user scikit-image "numpy<2"
docker exec -e ONEFLOW_MLIR_ENABLE_INFERENCE_OPTIMIZATION=0 ${{ env.CONTAINER_NAME }} python3 -m pytest -v onediff_diffusers_extensions/tests/test_lora.py --disable-warnings
- name: Install Requirements for WebUI
Expand Down
1 change: 0 additions & 1 deletion tests/comfy-docker-compose.yml
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,6 @@ services:
CI: "1"
SILICON_ONEDIFF_LICENSE_KEY: ${SILICON_ONEDIFF_LICENSE_KEY}
volumes:
- $HOME/test-container-cache-${CONTAINER_NAME}/dot-local:/root/.local
- $HOME/test-container-cache-${CONTAINER_NAME}/dot-cache:/root/.cache
- /share_nfs:/share_nfs:ro
- ${PWD}/${COMFYUI_SRC_DIR}:/app/ComfyUI
Expand Down
1 change: 0 additions & 1 deletion tests/diffusers-docker-compose.yml
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,6 @@ services:
CI: "1"
SILICON_ONEDIFF_LICENSE_KEY: ${SILICON_ONEDIFF_LICENSE_KEY}
volumes:
- $HOME/test-container-cache-${CONTAINER_NAME}/dot-local:/root/.local
- $HOME/test-container-cache-${CONTAINER_NAME}/dot-cache:/root/.cache
- /share_nfs:/share_nfs:ro
- ${SDXL_BASE}:/app/ComfyUI/models/checkpoints/sd_xl_base_1.0.safetensors:ro
Expand Down
1 change: 0 additions & 1 deletion tests/webui-docker-compose.yml
Original file line number Diff line number Diff line change
Expand Up @@ -46,7 +46,6 @@ services:
volumes:
- $HOME/test-container-cache-${CONTAINER_NAME}/dot-local:/root/.local
- $HOME/test-container-cache-${CONTAINER_NAME}/dot-cache:/root/.cache
- /share_nfs:/share_nfs:ro
- /share_nfs/onediff_ci/sd-webui/images:/share_nfs/onediff_ci/sd-webui/images:rw
Expand Down

0 comments on commit 73eaeb7

Please sign in to comment.