|
|
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="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", |
|
|
], |
|
|
) |
|
|
|
|
|
|
