/aiyabot

A neat Discord bot for Vladmandic Web UI

Primary LanguagePythonGNU General Public License v2.0GPL-2.0

AIYA

A Discord bot interface for Vladmandic Stable Diffusion SD.Next

Usage

To generate an image from text, use the /draw command and include your prompt as the query.

Currently supported options

  • negative prompts
  • swap model/checkpoint (see wiki)
  • sampling steps
  • width/height
  • CFG scale
  • sampling method
  • seed
  • Web UI styles
  • extra networks (hypernetwork, LoRA)
  • face restoration
  • high-res fix
  • CLIP skip
  • img2img
  • denoising strength
  • batch count

Bonus features

  • /settings command - set per-channel defaults for supported options (see Notes!):
    • also can set maximum steps limit and max batch count limit
    • refresh (update AIYA's options with any changes from Web UI)
  • /identify command - create a caption for your image.
  • /stats command - shows how many /draw commands have been used.
  • /info command - basic usage guide, other info, and download batch images.
  • /upscale command - resize your image.
  • buttons - certain outputs will contain buttons.
    • 🖋 - edit prompt, then generate a new image with same parameters.
    • 🎲 - randomize seed, then generate a new image with same parameters.
    • 📋 - view the generated image's information.
    • ⬆️ - upscale the generated image with defaults. Batch grids require use of the drop downs
    • ❌ - deletes the generated image.
  • dropdown menus - batch images produce two drop down menus for the first 25 images.
    • The first menu prompts the bot to send only the images that you select at single images
    • The second menu prompts the bot to upscale the selected image from the batch.
  • context menu options - commands you can try on any message.
    • Get Image Info - view information of an image generated by Stable Diffusion.
    • Quick Upscale - upscale an image without needing to set options.
    • Batch Download - download all images of a batch set without needing to specify batch_id and image_id
  • configuration file - can change some of AIYA's operational aspects.

Setup requirements

  • Set up Vladmandic's SD.Next.
  • Run the Web UI as local host with API (COMMANDLINE_ARGS= --api).
  • Clone this repo.
  • Create a file in your cloned repo called ".env", formatted like so:
# .env
TOKEN = put your bot token here
  • Run AIYA by running launch.bat (or launch.sh for Linux)

NOTE:

To use a Diffusers model, within models.csv you must enter the string found in System -> System info: image

image

Deploy with Docker

AIYA can be deployed using Docker.

The docker image supports additional configuration by adding environment variables or config file updates detailed in the wiki.

Docker run

docker run --name aiyabot --network=host --restart=always -e TOKEN=your_token_here -e TZ=America/New_York -v ./aiyabot/outputs:/app/outputs -v ./aiyabot/resources:/app/resources -d ghcr.io/kilvoctu/aiyabot:latest

Note the following environment variables work with the docker image:

  • TOKEN - [Required] Discord bot token.
  • URL - URL of the Web UI API. Defaults to http://localhost:7860.
  • TZ - Timezone for the container in the format America/New_York. Defaults to America/New_York
  • APIUSER - API username if required for your Web UI instance.
  • APIPASS - API password if required for your Web UI instance.
  • USER - Username if required for your Web UI instance.
  • PASS - Password if required for your Web UI instance.

Docker compose

  • Clone the repo and refer to the docker-compose.yml file in the deploy directory.
  • Rename the /deploy/.env.example file to .env and update the TOKEN variable with your bot token (and any other configuration as desired).
  • Run docker-compose up -d to start the bot.

Notes

Credits

AIYA only exists thanks to these awesome people: