from setuptools import setup, find_packages

setup(
    name='imaginairy',
    version='0.0.1',
    description='AI imagined images.',
    packages=find_packages("imaginairy"),
    install_requires=[
        '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",
    ],
)