File size: 1,417 Bytes
32e5a66 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 | 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"],
},
)
|