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.
|
|
|
|
from setuptools import setup, find_packages
|
|
|
|
|
|
|
|
|
|
setup(
|
|
|
|
|
name="imaginairy",
|
|
|
|
|
version="0.0.1",
|
|
|
|
|
description="AI imagined images.",
|
|
|
|
|
packages=find_packages(include=("imaginairy", "imaginairy.*")),
|
|
|
|
|
entry_points={
|
|
|
|
|
"console_scripts": ["imagine=imaginairy.cmd_wrap:imagine_cmd"],
|
|
|
|
|
},
|
|
|
|
|
package_data={"imaginairy": ["configs/*.yaml"]},
|
|
|
|
|
install_requires=[
|
|
|
|
|
"click",
|
|
|
|
|
"torch",
|
|
|
|
|
"numpy",
|
|
|
|
|
"tqdm",
|
|
|
|
|
"diffusers",
|
|
|
|
|
"imageio==2.9.0",
|
|
|
|
|
"pytorch-lightning==1.4.2",
|
|
|
|
|
"omegaconf==2.1.1",
|
|
|
|
|
"einops==0.3.0",
|
|
|
|
|
"transformers==4.19.2",
|
|
|
|
|
"torchmetrics==0.6.0",
|
|
|
|
|
"torchvision>=0.13.1",
|
|
|
|
|
"kornia==0.6",
|
|
|
|
|
"clip @ git+https://github.com/openai/CLIP.git@d50d76daa670286dd6cacf3bcd80b5e4823fc8e1#egg=clip",
|
|
|
|
|
],
|
|
|
|
|
)
|
