File size: 1,426 Bytes
13c5606 | 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 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 | #!/usr/bin/env python
import io
import os
import re
from datetime import datetime
from setuptools import find_packages, setup
def read(*names, **kwargs):
with io.open(os.path.join(os.path.dirname(__file__), *names),
encoding=kwargs.get("encoding", "utf8")) as fp:
return fp.read()
def find_version(*file_paths):
version_file = read(*file_paths)
version_match = re.search(r"^__version__ = ['\"]([^'\"]*)['\"]", version_file, re.M)
if version_match:
return version_match.group(1)
raise RuntimeError("Unable to find version string.")
VERSION = find_version('python', 'dinfer', '__init__.py')
if VERSION.endswith('dev'):
VERSION = VERSION + datetime.today().strftime('%Y%m%d')
requirements = [
'scipy',
'tqdm',
#'vllm==0.10.2',
#'sglang==0.5.3.post1',
]
extensions = []
cmdclass = {}
setup(
# Metadata
name='dinfer',
version=VERSION,
python_requires='>=3.6',
description='Package for dLLM inference',
long_description_content_type='text/markdown',
license='Apache-2.0',
# Package info
packages=find_packages(where="python", exclude=(
'tests',
)),
package_dir={"": "python"},
package_data={'': [os.path.join('datasets', 'dataset_checksums', '*.txt')]},
zip_safe=True,
include_package_data=True,
install_requires=requirements,
ext_modules=extensions,
cmdclass=cmdclass,
)
|