Fix versions

.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 \

.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: |

.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 \

README.md

@@ -298,12 +298,13 @@ This project leverages [timm](https://github.com/huggingface/pytorch-image-model
 
 ### BibTeX
 
-```bash
+```bibtex
 @misc{gc2025zeromodels,
   author = {Gitesh Chawda},
   title = {ZeroModels},
   year = {2025},
   publisher = {GitHub},
   journal = {GitHub repository},
   howpublished = {\url{https://github.com/IMvision12/ZeroModels}}
+}
 ```

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

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

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