mirror of
https://github.com/brycedrennan/imaginAIry
synced 2024-10-31 03:20:40 +00:00
33 lines
915 B
Python
33 lines
915 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=[
|
||
|
|
'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",
|
||
|
|
],
|
||
|
|
)
|