imaginAIry/setup.py
Bryce Drennan 31c2160e21
feature: prompt expansion (#51)
You can use `{}` to randomly pull values from lists.  A list of values separated by `|` and enclosed in `{ }` will be randomly drawn from in a non-repeating fashion. Values that are surrounded by `_ _` will pull from a phrase list of the same name.   Folders containing .txt phraselist files may be specified via
`--prompt_library_path`. The option may be specified multiple times.  Built-in categories:

      3d-term, adj-architecture, adj-beauty, adj-detailed, adj-emotion, adj-general, adj-horror, animal, art-movement,
      art-site, artist, artist-botanical, artist-surreal, aspect-ratio, bird, body-of-water, body-pose, camera-brand,
      camera-model, color, cosmic-galaxy, cosmic-nebula, cosmic-star, cosmic-term, dinosaur, eyecolor, f-stop,
      fantasy-creature, fantasy-setting, fish, flower, focal-length, food, fruit, games, gen-modifier, hair, hd,
      iso-stop, landscape-type, national-park, nationality, neg-weight, noun-beauty, noun-fantasy, noun-general,
      noun-horror, occupation, photo-term, pop-culture, pop-location, punk-style, quantity, rpg-item, scenario-desc,
      skin-color, spaceship, style, tree-species, trippy, world-heritage-site

   Examples:

   `imagine "a {red|black} dog" -r 2 --seed 0` will generate both "a red dog" and "a black dog"

   `imagine "a {_color_} dog" -r 4 --seed 0` will generate four, different colored dogs. The colors will eb pulled from an included
   phraselist of colors.

   `imagine "a {_spaceship_|_fruit_|hot air balloon}. low-poly" -r 4 --seed 0` will generate images of spaceships or fruits or a hot air balloon

   Credit to [noodle-soup-prompts](https://github.com/WASasquatch/noodle-soup-prompts/) where most, but not all, of the wordlists originate.
2022-10-08 18:34:35 -07:00

60 lines
1.7 KiB
Python

from setuptools import find_packages, setup
with open("README.md", "r", encoding="utf-8") as f:
readme = f.read()
setup(
name="imaginAIry",
author="Bryce Drennan",
# author_email="b r y p y d o t io",
version="2.3.1",
description="AI imagined images. Pythonic generation of stable diffusion images.",
long_description=readme,
long_description_content_type="text/markdown",
project_urls={
"Documentation": "https://github.com/brycedrennan/imaginAIry/blob/master/README.md",
"Source": "https://github.com/brycedrennan/imaginAIry",
},
packages=find_packages(include=("imaginairy", "imaginairy.*")),
entry_points={
"console_scripts": [
"imagine=imaginairy.cmds:imagine_cmd",
"aimg=imaginairy.cmds:aimg",
],
},
package_data={
"imaginairy": [
"configs/*.yaml",
"enhancers/wordlists/*.txt",
"vendored/clip/*.txt.gz",
"vendored/clipseg/*.pth",
"vendored/blip/configs/*.*",
"vendored/noodle_soup_prompts/*.*",
]
},
install_requires=[
"click",
"click-shell",
"protobuf != 3.20.2, != 3.19.5",
"fairscale>=0.4.4", # for vendored blip
"ftfy", # for vendored clip
"torch>=1.2.0",
"numpy",
"tqdm",
"diffusers",
"imageio==2.9.0",
"Pillow>=8.0.0",
"pytorch-lightning==1.4.2",
"omegaconf==2.1.1",
"einops==0.3.0",
"timm>=0.4.12", # for vendored blip
"torchdiffeq",
"transformers==4.19.2",
"torchmetrics==0.6.0",
"torchvision>=0.13.1",
"kornia==0.6",
"realesrgan",
"gfpgan>=1.3.7",
],
)