|
a |
|
b/setup.py |
|
|
1 |
import setuptools |
|
|
2 |
from setuptools.command.install import install |
|
|
3 |
from setuptools.command.develop import develop |
|
|
4 |
from setuptools.command.egg_info import egg_info |
|
|
5 |
|
|
|
6 |
with open("./README.md", "r") as fh: |
|
|
7 |
long_description = fh.read() |
|
|
8 |
|
|
|
9 |
setuptools.setup( |
|
|
10 |
name="medgpt", |
|
|
11 |
version="0.4", |
|
|
12 |
author="w-is-h", |
|
|
13 |
author_email="w.kraljevic@gmail.com", |
|
|
14 |
description="Temporal modeling of patients and diseases", |
|
|
15 |
long_description=long_description, |
|
|
16 |
long_description_content_type="text/markdown", |
|
|
17 |
url="https://github.com/w-is-h/medgpt", |
|
|
18 |
packages=['medgpt', 'medgpt.datasets', 'medgpt.metrics', 'medgpt.utils', |
|
|
19 |
'medgpt.models', 'medgpt.tokenizers'], |
|
|
20 |
install_requires=[ |
|
|
21 |
'datasets==2.15.0' |
|
|
22 |
'transformers==4.35.2', |
|
|
23 |
'flash-attn==2.3.6', |
|
|
24 |
], |
|
|
25 |
classifiers=[ |
|
|
26 |
"Programming Language :: Python :: 3", |
|
|
27 |
"License :: OSI Approved :: MIT License", |
|
|
28 |
"Operating System :: OS Independent", |
|
|
29 |
], |
|
|
30 |
) |
