imaginAIry/setup.py

38 lines
1.2 KiB
Python
Raw Normal View History

2022-09-11 20:58:14 +00:00
from setuptools import find_packages, setup
2022-09-08 03:59:30 +00:00
with open("README.md", "r", encoding="utf-8") as f:
readme = f.read()
2022-09-08 03:59:30 +00:00
setup(
name="imaginAIry",
author="Bryce Drennan",
author_email="b r y p y d o t io",
version="0.7.0",
description="AI imagined images. Pythonic generation of stable diffusion images.",
long_description=readme,
long_description_content_type="text/markdown",
packages=find_packages(include=("imaginairy", "imaginairy.*")),
2022-09-09 05:22:55 +00:00
entry_points={
"console_scripts": ["imagine=imaginairy.cmd_wrap:imagine_cmd"],
2022-09-09 05:22:55 +00:00
},
package_data={"imaginairy": ["configs/*.yaml"]},
2022-09-08 03:59:30 +00:00
install_requires=[
"click",
"torch",
"numpy",
"tqdm",
2022-09-11 07:35:57 +00:00
"diffusers",
2022-09-08 03:59:30 +00:00
"imageio==2.9.0",
"pytorch-lightning==1.4.2",
"omegaconf==2.1.1",
"einops==0.3.0",
"transformers==4.19.2",
"torchmetrics==0.6.0",
"torchvision>=0.13.1",
2022-09-08 03:59:30 +00:00
"kornia==0.6",
"clip @ git+https://github.com/openai/CLIP",
# k-diffusion for use with find_noise.py
# "k-diffusion@git+https://github.com/crowsonkb/k-diffusion.git@71ba7d6735e9cba1945b429a21345960eb3f151c#egg=k-diffusion",
2022-09-08 03:59:30 +00:00
],
)