File size: 1,452 Bytes
364cd6b |
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 |
from setuptools import setup, find_packages
from pathlib import Path
# Read README for long description
readme_file = Path(__file__).parent / "README.md"
long_description = readme_file.read_text(encoding="utf-8") if readme_file.exists() else ""
setup(
name="chest2vec",
version="0.6.0",
description="Section-aware embeddings for chest X-ray reports",
long_description=long_description,
long_description_content_type="text/markdown",
author="Chest2Vec Team",
url="https://github.com/chest2vec/chest2vec",
packages=find_packages(),
py_modules=["chest2vec"],
include_package_data=True,
package_data={"": ["__init__.py"]},
install_requires=[
"transformers==4.57.3",
"trl==0.9.3",
"deepspeed==0.16.9",
"peft",
"huggingface_hub",
"bitsandbytes",
"accelerate",
"numpy",
],
python_requires=">=3.8",
classifiers=[
"Development Status :: 4 - Beta",
"Intended Audience :: Developers",
"Intended Audience :: Science/Research",
"License :: OSI Approved :: Apache Software License",
"Programming Language :: Python :: 3",
"Programming Language :: Python :: 3.8",
"Programming Language :: Python :: 3.9",
"Programming Language :: Python :: 3.10",
"Programming Language :: Python :: 3.11",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
],
)
|