Skip to main content

A Stable Diffusion GUI

Project description

Banner Discord PyPi GitHub GitHub last commit GitHub issues GitHub closed issues GitHub pull requests GitHub closed pull requests


AI Runner

AI Runner is an interface which allows you to run open-source large language models (LLM) and AI image generators (Stable Diffusion) on your own hardware.

It is designed to be easy to use, with a simple and intuitive interface that allows you to run AI models without the need for a web server or cloud service.

It has been optimized for speed and efficiency, allowing you to generate images and have conversations with chatbots in real-time.

Stable Diffusion

img.png

Customizable Chatbots with Moods and Personalities

img_1.png


โญ Features

AI Runner is an AI interface which allows you to run open-source large language models (LLM) and AI image generators (Stable Diffusion) on your own hardware.

Feature Description
๐Ÿ—ฃ๏ธ LLMs and communication
Voice-based chatbot conversations Have conversations with a chatbot using your voice
Text-to-speech Convert text to spoken audio
Speech-to-text Convert spoken audio to text
Customizable chatbots with LLMs Generate text using large language models
RAG on local documents and websites Interact with your local documents using an LLM
๐ŸŽจ Image Generation
Stable Diffusion (all versions) Generate images using Stable Diffusion
Drawing tools Turn sketches into art
Text-to-Image Generate images from textual descriptions
Image-to-Image Generate images based on input images
๐Ÿ–ผ๏ธ Image Manipulation
Inpaint and Outpaint Modify parts of an image while maintaining context
Controlnet Control image generation with additional input
LoRA Efficiently fine-tune models with LoRA
Textual Embeddings Use textual embeddings for image generation control
Image Filters Blur, film grain, pixel art and more
๐Ÿ”ง Utility
Run offline, locally Run on your own hardware without internet
Fast generation Generate images in ~2 seconds (RTX 2080s)
Run multiple models at once Utilize multiple models simultaneously
Dark mode Comfortable viewing experience in low-light environments
Infinite scrolling canvas Seamlessly scroll through generated images
NSFW filter toggle Help control the visibility of NSFW content
NSFW guardrails toggle Help prevent generation of LLM harmful content
Fully customizable Easily adjust all parameters
Fast load time, responsive interface Enjoy a smooth and responsive user experience
Pure python No reliance on a webserver, pure python implementation

๐Ÿ’ป System Requirements

Minimum system requirements

  • OS: Linux
  • Processor: Intel i5 or equivalent
  • Memory: 16 GB RAM
  • Graphics: 2080s RTX or higher
  • Network: Broadband Internet connection required for setup
  • Storage: 130 GB available space

Recommended system specs

  • OS: Linux
  • Processor: Intel i7 or equivalent
  • Memory: 30 GB RAM
  • Graphics: 4090 RTX or higher
  • Network: Broadband Internet connection required for setup
  • Storage: 130 GB available space

๐Ÿš€ Installation

There are several ways to get started with AI Runner such as packaged, from source and as a library.

Detailed packaging and installation instructions can be found in the wiki.

Install as a library

Installation

pip install airunner

Running

airunner

Models

AI Runner installs all of the models required to run a chatbot with text-to-speech and speech-to-text capabilities, as well as the core models required for Stable Diffusion. However, you must supply your own art generator models.

You can download models from Huggingface.co or civitai.com.

The supported Stable Diffusion models are:

  • SD 1.5
  • SDXL 1.0
  • SDXL Turbo

Models must be placed in their respective directories in the airunner directory.

~/.local/share/airunner
โ”œโ”€โ”€ art
โ”‚   โ”œโ”€โ”€ models
โ”‚   โ”‚   โ”œโ”€โ”€ SD 1.5
โ”‚   โ”‚   โ”‚   โ”œโ”€โ”€ lora
โ”‚   โ”‚   โ”‚   โ””โ”€โ”€ embeddings
โ”‚   โ”‚   โ”œโ”€โ”€ SDXL 1.0
โ”‚   โ”‚   โ”‚   โ”œโ”€โ”€ lora
โ”‚   โ”‚   โ”‚   โ””โ”€โ”€ embeddings
โ”‚   โ”‚   โ””โ”€โ”€ SDXL Turbo
โ”‚   โ”‚       โ”œโ”€โ”€ lora
โ”‚   โ”‚       โ””โ”€โ”€ embeddings

Unit tests

Run all unit tests

python -m unittest discover -s src/airunner/tests

Run a single unit tests python -m unittest src/airunner/tests/<file_name>

Example

python -m unittest src/airunner/tests/test_prompt_weight_convert.py

Privacy and Security

Although AI Runner v3.0 is built with Huggingface libraries, we have taken care to strip the application of any telemetry or tracking features.


Internet access

Only the setup wizard needs access to the internet in order to download the required models.

For more information see the Darklock and Facehuggershield libraries.


Disc access

Write access for the transformers library has been disabled, preventing it from creating a huggingface cache directory at runtime.

The application itself may still access the disc for reading and writing, however we have restricted reads and writes to the user provided airunner directory (by default this is located at ~/.local/share/airunner).

All other attempts to access the disc are blocked and logged for your review.

For more information see src/security/restrict_os_access.py.


Huggingface Hub

Huggingface Hub contains telemetry and tracking features that have been completely disabled in AI Runner.

The security measures taken for this library are as follows

  • Prevented from accessing the internet
  • Prevented from accessing the disc
  • All environment variables set for maximum security
  • All telemetry disabled

See Facehuggershield for more information.

Project details


Release history Release notifications | RSS feed

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

airunner-3.1.1.tar.gz (1.3 MB view details)

Uploaded Source

Built Distribution

airunner-3.1.1-py3-none-any.whl (1.6 MB view details)

Uploaded Python 3

File details

Details for the file airunner-3.1.1.tar.gz.

File metadata

  • Download URL: airunner-3.1.1.tar.gz
  • Upload date:
  • Size: 1.3 MB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/5.1.1 CPython/3.9.20

File hashes

Hashes for airunner-3.1.1.tar.gz
Algorithm Hash digest
SHA256 7aa8e127e3fda1d1f45ccdd3e6f17734672b5e77996354a7ff8be6268da6d427
MD5 fdaa549fd519b730f76eefe23515f97d
BLAKE2b-256 2790c6d0d3664d604f9c8e67c57debf5c67a6b771aa7e31c913d7ed17e4ed28f

See more details on using hashes here.

File details

Details for the file airunner-3.1.1-py3-none-any.whl.

File metadata

  • Download URL: airunner-3.1.1-py3-none-any.whl
  • Upload date:
  • Size: 1.6 MB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/5.1.1 CPython/3.9.20

File hashes

Hashes for airunner-3.1.1-py3-none-any.whl
Algorithm Hash digest
SHA256 a2726bdbdd9137635011516f453f3280da70915d62e5560be2c08ad61cbd8e0b
MD5 61e54319cce95d425c61af3dbf98b47d
BLAKE2b-256 f75ce2bc5a222b834cf2af1b0bb6db18ada39431057c68060750b6da695dde44

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page