from setuptools import setup, find_packages

setup(
    name='imaginarium',
    version='0.2',
    packages=find_packages(),
    install_requires=[
        'torch',
        'numpy',
        'tqdm',
        'redis',
        'pandas',
        "clean-fid",
        "colab-convert",
        "einops",
        "ftfy",
        "ipython",
        "ipywidgets",
        "jsonmerge",
        "jupyterlab",
        "jupyter_http_over_ws",
        "kornia",
        "matplotlib",
        "notebook",
        "numexpr",
        "omegaconf",
        "opencv-python",
        "pytorch_lightning==1.7.7",
        "resize-right",
        "scikit-image==0.19.3",
        "scikit-learn",
        "timm",
        "torchdiffeq",
        "transformers==4.19.2",
        "safetensors",
        "albumentations",
        "more_itertools",
        "devtools",
        "validators",
        "numpngw",
        "open-clip-torch",
        "torchsde",
        "ninja",
        "pydantic",
        "screeninfo"
    ],
)
