Unverified Commit 24281f80 authored by Aryan's avatar Aryan Committed by GitHub
Browse files

`make deps_table_update` to fix CI tests (#9720)



* update

* dummy change to trigger CI; will revert

* no deps peft

* np deps

* todo

---------
Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
parent 2a1d2f62
...@@ -92,12 +92,14 @@ jobs: ...@@ -92,12 +92,14 @@ jobs:
run: | run: |
python -m venv /opt/venv && export PATH="/opt/venv/bin:$PATH" python -m venv /opt/venv && export PATH="/opt/venv/bin:$PATH"
python -m uv pip install -e [quality,test] python -m uv pip install -e [quality,test]
# TODO (sayakpaul, DN6): revisit `--no-deps`
if [ "${{ matrix.lib-versions }}" == "main" ]; then if [ "${{ matrix.lib-versions }}" == "main" ]; then
python -m pip install -U peft@git+https://github.com/huggingface/peft.git python -m pip install -U peft@git+https://github.com/huggingface/peft.git --no-deps
python -m uv pip install -U transformers@git+https://github.com/huggingface/transformers.git python -m uv pip install -U transformers@git+https://github.com/huggingface/transformers.git --no-deps
pip uninstall accelerate -y && python -m uv pip install -U accelerate@git+https://github.com/huggingface/accelerate.git pip uninstall accelerate -y && python -m uv pip install -U accelerate@git+https://github.com/huggingface/accelerate.git --no-deps
else else
python -m uv pip install -U peft transformers accelerate python -m uv pip install -U peft --no-deps
python -m uv pip install -U transformers accelerate --no-deps
fi fi
- name: Environment - name: Environment
......
...@@ -38,7 +38,7 @@ deps = { ...@@ -38,7 +38,7 @@ deps = {
"regex": "regex!=2019.12.17", "regex": "regex!=2019.12.17",
"requests": "requests", "requests": "requests",
"tensorboard": "tensorboard", "tensorboard": "tensorboard",
"torch": "torch>=1.4", "torch": "torch>=1.4,<2.5.0",
"torchvision": "torchvision", "torchvision": "torchvision",
"transformers": "transformers>=4.41.2", "transformers": "transformers>=4.41.2",
"urllib3": "urllib3<=2.0.0", "urllib3": "urllib3<=2.0.0",
......
...@@ -1060,7 +1060,6 @@ class StableDiffusionPipeline( ...@@ -1060,7 +1060,6 @@ class StableDiffusionPipeline(
do_denormalize = [True] * image.shape[0] do_denormalize = [True] * image.shape[0]
else: else:
do_denormalize = [not has_nsfw for has_nsfw in has_nsfw_concept] do_denormalize = [not has_nsfw for has_nsfw in has_nsfw_concept]
image = self.image_processor.postprocess(image, output_type=output_type, do_denormalize=do_denormalize) image = self.image_processor.postprocess(image, output_type=output_type, do_denormalize=do_denormalize)
# Offload all models # Offload all models
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment