from setuptools import find_packages, setup with open("README.md", "r", encoding="utf-8") as f: readme = f.read() setup( name="imaginAIry", author="Bryce Drennan", # author_email="b r y p y d o t io", version="1.2.0", description="AI imagined images. Pythonic generation of stable diffusion images.", long_description=readme, long_description_content_type="text/markdown", project_urls={ "Documentation": "https://github.com/brycedrennan/imaginAIry/blob/master/README.md", "Source": "https://github.com/brycedrennan/imaginAIry", }, packages=find_packages(include=("imaginairy", "imaginairy.*")), entry_points={ "console_scripts": ["imagine=imaginairy.cmds:imagine_cmd"], }, package_data={"imaginairy": ["configs/*.yaml", "vendored/clip/*.txt.gz", "vendored/clipseg/*.pth"]}, install_requires=[ "click", "protobuf != 3.20.2, != 3.19.5", "ftfy", # for vendored clip "torch>=1.2.0", "numpy", "tqdm", "diffusers", "imageio==2.9.0", "pytorch-lightning==1.4.2", "omegaconf==2.1.1", "einops==0.3.0", "torchdiffeq", "transformers==4.19.2", "torchmetrics==0.6.0", "torchvision>=0.13.1", "kornia==0.6", "realesrgan", "gfpgan>=1.3.7", ], )