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=[
|
2022-09-09 04:51:25 +00:00
|
|
|
'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",
|
|
|
|
|
],
|
|
|
|
|
)
|
