2022-09-11 20:58:14 +00:00
|
|
|
from setuptools import find_packages, setup
|
2022-09-08 03:59:30 +00:00
|
|
|
|
2022-09-12 04:36:14 +00:00
|
|
|
with open("README.md", "r", encoding="utf-8") as f:
|
|
|
|
|
readme = f.read()
|
|
|
|
|
|
2022-09-08 03:59:30 +00:00
|
|
|
setup(
|
2022-09-12 04:36:14 +00:00
|
|
|
name="imaginAIry",
|
|
|
|
|
author="Bryce Drennan",
|
2022-09-12 04:47:14 +00:00
|
|
|
# author_email="b r y p y d o t io",
|
2022-10-04 13:21:37 +00:00
|
|
|
version="2.2.1",
|
2022-09-12 04:36:14 +00:00
|
|
|
description="AI imagined images. Pythonic generation of stable diffusion images.",
|
|
|
|
|
long_description=readme,
|
|
|
|
|
long_description_content_type="text/markdown",
|
2022-09-12 04:47:14 +00:00
|
|
|
project_urls={
|
|
|
|
|
"Documentation": "https://github.com/brycedrennan/imaginAIry/blob/master/README.md",
|
|
|
|
|
"Source": "https://github.com/brycedrennan/imaginAIry",
|
|
|
|
|
},
|
2022-09-10 07:32:31 +00:00
|
|
|
packages=find_packages(include=("imaginairy", "imaginairy.*")),
|
2022-09-09 05:22:55 +00:00
|
|
|
entry_points={
|
2022-09-20 04:15:38 +00:00
|
|
|
"console_scripts": [
|
|
|
|
|
"imagine=imaginairy.cmds:imagine_cmd",
|
|
|
|
|
"aimg=imaginairy.cmds:aimg",
|
|
|
|
|
],
|
2022-09-09 05:22:55 +00:00
|
|
|
},
|
2022-09-18 13:07:07 +00:00
|
|
|
package_data={
|
|
|
|
|
"imaginairy": [
|
|
|
|
|
"configs/*.yaml",
|
|
|
|
|
"vendored/clip/*.txt.gz",
|
|
|
|
|
"vendored/clipseg/*.pth",
|
2022-09-21 00:05:11 +00:00
|
|
|
"vendored/blip/configs/*.*",
|
2022-09-18 13:07:07 +00:00
|
|
|
]
|
|
|
|
|
},
|
2022-09-08 03:59:30 +00:00
|
|
|
install_requires=[
|
2022-09-10 07:32:31 +00:00
|
|
|
"click",
|
2022-09-24 08:56:19 +00:00
|
|
|
"click-shell",
|
2022-09-16 15:18:50 +00:00
|
|
|
"protobuf != 3.20.2, != 3.19.5",
|
2022-09-20 04:15:38 +00:00
|
|
|
"fairscale>=0.4.4", # for vendored blip
|
2022-09-12 05:22:43 +00:00
|
|
|
"ftfy", # for vendored clip
|
2022-09-13 07:27:53 +00:00
|
|
|
"torch>=1.2.0",
|
2022-09-10 07:32:31 +00:00
|
|
|
"numpy",
|
|
|
|
|
"tqdm",
|
2022-09-11 07:35:57 +00:00
|
|
|
"diffusers",
|
2022-09-08 03:59:30 +00:00
|
|
|
"imageio==2.9.0",
|
2022-09-21 04:15:19 +00:00
|
|
|
"Pillow>=8.0.0",
|
2022-09-08 03:59:30 +00:00
|
|
|
"pytorch-lightning==1.4.2",
|
|
|
|
|
"omegaconf==2.1.1",
|
|
|
|
|
"einops==0.3.0",
|
2022-09-20 04:15:38 +00:00
|
|
|
"timm>=0.4.12", # for vendored blip
|
2022-09-14 07:40:25 +00:00
|
|
|
"torchdiffeq",
|
2022-09-08 03:59:30 +00:00
|
|
|
"transformers==4.19.2",
|
|
|
|
|
"torchmetrics==0.6.0",
|
2022-09-10 07:32:31 +00:00
|
|
|
"torchvision>=0.13.1",
|
2022-09-08 03:59:30 +00:00
|
|
|
"kornia==0.6",
|
2022-09-13 07:27:53 +00:00
|
|
|
"realesrgan",
|
|
|
|
|
"gfpgan>=1.3.7",
|
2022-09-08 03:59:30 +00:00
|
|
|
],
|
|
|
|
|
)
|
