imaginAIry/setup.py

37 lines
1.0 KiB
Python
Raw Normal View History

2022-09-08 03:59:30 +00:00
from setuptools import setup, find_packages
setup(
name='imaginairy',
version='0.0.1',
description='AI imagined images.',
packages=find_packages("imaginairy"),
2022-09-09 05:22:55 +00:00
entry_points={
'console_scripts': ['imagine=imaginairy.cmds:imagine_cmd'],
},
2022-09-08 03:59:30 +00:00
install_requires=[
'click',
2022-09-08 03:59:30 +00:00
'torch',
'numpy',
'tqdm',
2022-09-09 05:22:55 +00:00
# "albumentations==0.4.3",
2022-09-08 03:59:30 +00:00
"diffusers",
# opencv-python==4.1.2.30
"pudb==2019.2",
2022-09-09 05:22:55 +00:00
# "invisible-watermark",
2022-09-08 03:59:30 +00:00
"imageio==2.9.0",
"imageio-ffmpeg==0.4.2",
"pytorch-lightning==1.4.2",
"omegaconf==2.1.1",
"test-tube>=0.7.5",
"streamlit>=0.73.1",
"einops==0.3.0",
"torch-fidelity==0.3.0",
"transformers==4.19.2",
"torchmetrics==0.6.0",
"kornia==0.6",
"realesrgan",
"-e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers",
"-e git+https://github.com/openai/CLIP.git@main#egg=clip",
],
)