| import setuptools | |
| setuptools.setup( | |
| name="unilm", # your choice, but usually same as the package | |
| version="0.1.0", # obligatory | |
| packages=["dit"], # name of the folder relative to this file where the source code lives | |
| install_requires=[ | |
| "setuptools==58.2.0", | |
| "timm==0.5.4", | |
| "Pillow", | |
| "blobfile", | |
| "mypy", | |
| "numpy", | |
| "pytest", | |
| "requests", | |
| "einops", | |
| "tensorboardX", | |
| "deepspeed==0.4.0", | |
| "scipy", | |
| "opencv-python", | |
| "datasets", | |
| "huggingface_hub", | |
| "ipywidgets", | |
| "shapely==1.8", | |
| ], # your dependencies, if you have any | |
| entry_points = { # this here is the magic that binds your function into a callable script | |
| "console_scripts": ["train=dit.train_net:entry_point"], | |
| } | |
| ) |