Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 1 addition & 2 deletions .github/workflows/test_models.yml
Original file line number Diff line number Diff line change
Expand Up @@ -55,9 +55,8 @@ jobs:
if [ "${{ matrix.executorch-version }}" == "nightly" ]; then
python install_dev.py
else
pip install '.[tests]'
pip install '.[dev]'
pip install executorch==${{ matrix.executorch-version }}
pip install git+https://github.com/pytorch-labs/tokenizers
fi
pip list
- name: Run tests
Expand Down
4 changes: 2 additions & 2 deletions install_dev.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@ def install_torch_nightly_deps():
"""Install torch related dependencies from pinned nightly"""
EXECUTORCH_NIGHTLY_VERSION = "dev20250916"
TORCHAO_NIGHTLY_VERSION = "dev20250916"
# Torch nightly is aligned with pinned nightly in https://github.com/pytorch/executorch/blob/main/install_requirements.py#L74
# Torch nightly is aligned with pinned nightly in https://github.com/pytorch/executorch/blob/main/torch_pin.py#L2
TORCH_NIGHTLY_VERSION = "dev20250916"
subprocess.check_call(
[
Expand Down Expand Up @@ -43,7 +43,7 @@ def install_dep_from_source():
"-m",
"pip",
"install",
"git+https://github.com/pytorch-labs/tokenizers@fc32028858020c4fcafe37aaaeaf5d1b480336a2#egg=pytorch-tokenizers",
"git+https://github.com/pytorch-labs/tokenizers@3aada3fe28c945d14d5ec62254eb56ccdf10eb11#egg=pytorch-tokenizers",
]
)

Expand Down
83 changes: 78 additions & 5 deletions pyproject.toml
Original file line number Diff line number Diff line change
@@ -1,14 +1,83 @@
[project]
name = "optimum-executorch"
dynamic = ["version"]
description = "Optimum Executorch is an interface between the Hugging Face libraries and ExecuTorch"
readme = { file = "README.md", content-type = "text/markdown" }
license = { text = "Apache" }
authors = [
{ name = "HuggingFace Inc. Special Ops Team", email = "[email protected]" },
]
requires-python = ">=3.10.0"
keywords = ["transformers", "quantization", "inference", "executorch"]
classifiers = [
"Development Status :: 2 - Pre-Alpha",
"License :: OSI Approved :: Apache Software License",
"Intended Audience :: Developers",
"Intended Audience :: Education",
"Intended Audience :: Science/Research",
"Operating System :: OS Independent",
"Programming Language :: Python :: 3",
"Programming Language :: Python :: 3.10",
"Programming Language :: Python :: 3.11",
"Programming Language :: Python :: 3.12",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
]

dependencies = [
"optimum~=1.24",
"executorch>=1.0.0",
"transformers==4.56.1",
"pytorch-tokenizers>=1.0.1",
"accelerate>=0.26.0",
]

[project.optional-dependencies]
dev = [
"accelerate>=0.26.0",
"coremltools>=8.2.0",
"datasets==3.6.0",
"parameterized",
"pytest",
"safetensors",
"sentencepiece",
"numba!=0.58.0",
"librosa",
"soundfile",
"tiktoken",
"black~=23.1",
"ruff==0.4.4",
]

[project.urls]
Homepage = "https://github.com/huggingface/optimum-executorch"

# ---- setuptools config ----

[tool.setuptools]
# Equivalent of include_package_data=True
include-package-data = true

[tool.setuptools.packages.find]
# Mirrors find_namespace_packages(include=["optimum*"])
include = ["optimum*"]
namespaces = true

[tool.setuptools.dynamic]
# Pull version from the Python attribute
version = { attr = "optimum.executorch.version.__version__" }

# ---- your existing tool configs (kept, but one tweak suggested) ----

[tool.black]
line-length = 119
target-version = ['py37']
# Recommended to match your supported interpreters:
target-version = ["py310", "py311", "py312"]

[tool.ruff]
# Never enforce `E501` (line length violations).
ignore = ["C901", "E501", "E741", "W605"]
select = ["C", "E", "F", "I", "W"]
line-length = 119

# Ignore import violations in all `__init__.py` files.
[tool.ruff.per-file-ignores]
"__init__.py" = ["E402", "F401", "F403", "F811"]

Expand All @@ -18,6 +87,10 @@ known-first-party = ["optimum"]

[tool.pytest.ini_options]
markers = [
"run_slow",
"portable",
"run_slow",
"portable",
]

[build-system]
requires = ["setuptools >= 77.0.3", "wheel"]
build-backend = "setuptools.build_meta"
75 changes: 0 additions & 75 deletions setup.py

This file was deleted.

Loading