You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
34 lines
932 B
Python
34 lines
932 B
Python
from setuptools import setup, find_packages
|
|
|
|
setup(
|
|
name='imaginairy',
|
|
version='0.0.1',
|
|
description='AI imagined images.',
|
|
packages=find_packages("imaginairy"),
|
|
install_requires=[
|
|
'click',
|
|
'torch',
|
|
'numpy',
|
|
'tqdm',
|
|
"albumentations==0.4.3",
|
|
"diffusers",
|
|
# opencv-python==4.1.2.30
|
|
"pudb==2019.2",
|
|
"invisible-watermark",
|
|
"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",
|
|
],
|
|
)
|