entropack/pyproject.toml at main · modelscope/entropack · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
[build-system]
requires = ["setuptools>=77.0.3", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "entropack"
version = "0.1.1"
description = "EntroPack: general-purpose tensor compression for PyTorch with lossless and rate-controlled lossy schemes and GPU encoding and decoding."
readme = "README.md"
requires-python = ">=3.10"
license = "Apache-2.0"
license-files = ["LICENSE"]
authors = [{ name = "DiffSynth-Studio" }]

dependencies = [
    "torch>=2.10",
    "numpy>=1.23",
    "dahuffman>=0.4",
]

[project.optional-dependencies]
cuda13 = ["cupy-cuda13x>=14"]
cuda12 = ["cupy-cuda12x>=14"]

[tool.setuptools.packages.find]
include = ["entropack*"]

[tool.setuptools.package-data]
"entropack.schemes.dfloat11" = ["*.cu"]
"entropack.schemes.tile_ans" = ["*.cu", "*.cuh"]
"entropack.schemes.lattice_rans" = ["*.cu"]

[tool.pytest.ini_options]
testpaths = ["tests"]