From 8425ccfb0a61962efd90cdf317d45d5e763905fb Mon Sep 17 00:00:00 2001 From: IMvision12 Date: Fri, 18 Sep 2026 15:42:51 -0700 Subject: [PATCH] Fix versions --- .github/workflows/cross_backend_parity.yml | 10 +++--- .github/workflows/release.yml | 2 +- .github/workflows/test_code.yml | 40 +++++++++++----------- README.md | 3 +- dev-requirements.txt | 31 ++++++++++++----- pyproject.toml | 2 +- requirements.txt | 9 ++--- 7 files changed, 57 insertions(+), 40 deletions(-) diff --git a/.github/workflows/cross_backend_parity.yml b/.github/workflows/cross_backend_parity.yml index 8f263014..1f0433ca 100644 --- a/.github/workflows/cross_backend_parity.yml +++ b/.github/workflows/cross_backend_parity.yml @@ -59,15 +59,15 @@ jobs: - name: Install dependencies run: | python -m pip install --upgrade pip setuptools - pip install -U keras>=3.15.0 pytest + pip install -U "keras>=3.15.0" pytest # CPU-only torch wheel (the default bundles CUDA/Triton which segfaults on # CUDA-less Linux runners via torchvision -> torch._dynamo). pip install --index-url https://download.pytorch.org/whl/cpu \ - torch>=2.5.0 torchvision - pip install jax[cpu] tensorflow-cpu>=2.19.0 - pip install sentencepiece matplotlib>=3.8.0 pillow + "torch>=2.5.0" torchvision + pip install "jax[cpu]" "tensorflow-cpu>=2.19.0" + pip install sentencepiece "matplotlib>=3.8.0" pillow pip install huggingface_hub safetensors - pip install -e . + pip install . - name: Run cross-backend parity run: | pytest tests/integration/test_cross_backend_parity.py \ diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml index 2b9a25af..4ee937f2 100644 --- a/.github/workflows/release.yml +++ b/.github/workflows/release.yml @@ -69,7 +69,7 @@ jobs: - name: Install dependencies run: | pip install -r requirements.txt --progress-bar off --upgrade - pip install -e ".[tests]" --progress-bar off --upgrade + pip install ".[test]" --progress-bar off --upgrade - name: Build wheels shell: bash run: | diff --git a/.github/workflows/test_code.yml b/.github/workflows/test_code.yml index a2a9f0cd..405b5b11 100644 --- a/.github/workflows/test_code.yml +++ b/.github/workflows/test_code.yml @@ -13,7 +13,7 @@ jobs: - uses: actions/setup-python@v5 with: python-version: '3.11' - - run: pip install ruff + - run: pip install ruff==0.16.1 - run: ruff check . - run: ruff format --check . @@ -54,18 +54,18 @@ jobs: - name: Install dependencies run: | python -m pip install --upgrade pip setuptools - pip install -U keras>=3.15.0 pytest pytest-cov + pip install -U "keras>=3.15.0" pytest pytest-cov # CPU-only torch wheel: the default PyPI wheel bundles # CUDA/Triton which segfaults during `import triton.knobs` # on CUDA-less Linux runners (gets pulled in transitively by # torchvision -> torch._dynamo). pip install --index-url https://download.pytorch.org/whl/cpu \ - torch>=2.5.0 torchvision - pip install jax[cpu] tensorflow-cpu>=2.19.0 - pip install sentencepiece matplotlib>=3.8.0 pillow + "torch>=2.5.0" torchvision + pip install "jax[cpu]" "tensorflow-cpu>=2.19.0" + pip install sentencepiece "matplotlib>=3.8.0" pillow pip install huggingface_hub safetensors pip install "transformers>=5.0.0" - pip install -e . + pip install . - name: Run all tests run: | pytest tests/integration/test_backend_compatibility.py \ @@ -105,11 +105,11 @@ jobs: - name: Install dependencies run: | python -m pip install --upgrade pip setuptools - pip install -U keras>=3.15.0 pytest - pip install --index-url https://download.pytorch.org/whl/cpu torch>=2.5.0 - pip install sentencepiece huggingface_hub matplotlib>=3.8.0 pillow + pip install -U "keras>=3.15.0" pytest + pip install --index-url https://download.pytorch.org/whl/cpu "torch>=2.5.0" + pip install sentencepiece huggingface_hub "matplotlib>=3.8.0" pillow pip install "transformers>=5.0.0" - pip install -e . + pip install . - name: Run tokenizer parity tests run: | pytest tests/processors/test_keras_tokenizers_vs_hf.py \ @@ -139,14 +139,14 @@ jobs: - name: Install dependencies run: | python -m pip install --upgrade pip setuptools - pip install -U keras>=3.15.0 pytest + pip install -U "keras>=3.15.0" pytest pip install --index-url https://download.pytorch.org/whl/cpu \ - torch>=2.5.0 torchvision - pip install jax[cpu] tensorflow-cpu>=2.19.0 - pip install sentencepiece matplotlib>=3.8.0 pillow + "torch>=2.5.0" torchvision + pip install "jax[cpu]" "tensorflow-cpu>=2.19.0" + pip install sentencepiece "matplotlib>=3.8.0" pillow pip install huggingface_hub safetensors pip install "transformers>=5.0.0" - pip install -e . + pip install . - name: Run image processor parity tests run: | pytest tests/processors/test_keras_image_processors_vs_hf.py \ @@ -176,14 +176,14 @@ jobs: - name: Install dependencies run: | python -m pip install --upgrade pip setuptools - pip install -U keras>=3.15.0 pytest + pip install -U "keras>=3.15.0" pytest pip install --index-url https://download.pytorch.org/whl/cpu \ - torch>=2.5.0 torchvision - pip install jax[cpu] tensorflow-cpu>=2.19.0 - pip install sentencepiece matplotlib>=3.8.0 pillow + "torch>=2.5.0" torchvision + pip install "jax[cpu]" "tensorflow-cpu>=2.19.0" + pip install sentencepiece "matplotlib>=3.8.0" pillow pip install huggingface_hub safetensors pip install "transformers>=5.0.0" - pip install -e . + pip install . - name: Run processor parity tests run: | pytest tests/processors/test_keras_processors_vs_hf.py \ diff --git a/README.md b/README.md index c8e19aef..3727ecc0 100644 --- a/README.md +++ b/README.md @@ -298,7 +298,7 @@ This project leverages [timm](https://github.com/huggingface/pytorch-image-model ### BibTeX -```bash +```bibtex @misc{gc2025zeromodels, author = {Gitesh Chawda}, title = {ZeroModels}, @@ -306,4 +306,5 @@ This project leverages [timm](https://github.com/huggingface/pytorch-image-model publisher = {GitHub}, journal = {GitHub repository}, howpublished = {\url{https://github.com/IMvision12/ZeroModels}} +} ``` diff --git a/dev-requirements.txt b/dev-requirements.txt index ffdbf838..edebacd1 100644 --- a/dev-requirements.txt +++ b/dev-requirements.txt @@ -1,12 +1,27 @@ +tensorflow-cpu>=2.19.0 + +--extra-index-url https://download.pytorch.org/whl/cpu +torch>=2.5.0 +torchvision>=0.20.0 + +jax[cpu] + keras>=3.15.0 -torch -torchvision -tensorflow-cpu>=2.18.0 -jax -ruff -pre-commit +huggingface_hub +safetensors +numpy +pillow +matplotlib>=3.8.0 +tokenizers +sentencepiece + timm transformers>=5.0.0 -accelerate +diffusers + pytest -sentencepiece +pytest-cov +requests + +pre-commit +ruff==0.16.1 diff --git a/pyproject.toml b/pyproject.toml index 32cbfcd5..a405f764 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -74,7 +74,7 @@ conversion = [ "torch>=2.5.0", "torchvision>=0.20.0", "timm", - "transformers", + "transformers>=5.0.0", "diffusers", ] # Developer tooling: the pre-commit hooks (ruff lint + format, whitespace/EOF diff --git a/requirements.txt b/requirements.txt index aed7976f..e009250d 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,13 +3,14 @@ tensorflow-cpu>=2.19.0 --extra-index-url https://download.pytorch.org/whl/cpu torch>=2.5.0 torchvision>=0.20.0 -sentencepiece -matplotlib>=3.8.0 -Pillow>=10.0.0 jax[cpu] keras>=3.15.0 - huggingface_hub safetensors +numpy +pillow +matplotlib>=3.8.0 +tokenizers +sentencepiece