| from setuptools import setup, find_packages |
|
|
| with open("README.md", "r", encoding="utf-8") as fh: |
| long_description = fh.read() |
|
|
| setup( |
| name="versatile_diffusion", |
| version="1.0.0", |
| author="Reese Kneeland", |
| description="An installable version of the versatile diffusion package", |
| long_description=long_description, |
| long_description_content_type="text/markdown", |
| url="https://github.com/reesekneeland/Versatile-Diffusion", |
| packages=find_packages(), |
| classifiers=[ |
| "Programming Language :: Python :: 3", |
| "License :: OSI Approved :: MIT License", |
| "Operating System :: OS Independent", |
| ], |
| python_requires='>=3.7', |
| install_requires=[ |
| "matplotlib", |
| "pyyaml", |
| "numpy", |
| "opencv-python", |
| "easydict==1.9", |
| "scikit-image", |
| "tensorboardx==2.1", |
| "tensorboard", |
| "lpips==0.1.3", |
| "fsspec", |
| "tqdm", |
| "transformers", |
| "torchmetrics==1.3.0.post0", |
| "einops", |
| "omegaconf", |
| "open_clip_torch==2.0.2", |
| "webdataset", |
| "huggingface-hub", |
| "gradio==3.17.1", |
| ], |
| include_package_data=True, |
| package_data={ |
| "versatile_diffusion": ["configs/model/*.yaml"], |
| "versatile_diffusion": ["lib/model_zoo/optimus_models/vocab/*.json"], |
| "versatile_diffusion": ["lib/model_zoo/optimus_models/vocab/*.txt"], |
| }, |
| ) |
|
|
