| from setuptools import setup, find_packages |
| from pathlib import Path |
|
|
| |
| 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="4.0.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", |
| ], |
| ) |
|
|
