mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2024-11-25 09:19:23 +01:00
91 lines
4.0 KiB
Markdown
91 lines
4.0 KiB
Markdown
## Description:
|
|
TL;DR: Lets the bot answer you with a picture!
|
|
|
|
Stable Diffusion API pictures for TextGen, v.1.2.0
|
|
An extension to [oobabooga's textgen-webui](https://github.com/oobabooga/text-generation-webui) allowing you to receive pics generated by [Automatic1111's SD-WebUI API](https://github.com/AUTOMATIC1111/stable-diffusion-webui)
|
|
|
|
<details>
|
|
<summary>Interface overview</summary>
|
|
|
|
![Interface](https://raw.githubusercontent.com/Brawlence/SD_api_pics/main/illust/Interface.jpg)
|
|
|
|
</details>
|
|
|
|
Load it in the `--chat` mode with `--extension sd_api_pictures` alongside `send_pictures`
|
|
(it's not really required, but completes the picture, *pun intended*).
|
|
|
|
|
|
## History
|
|
|
|
Consider the version included with [oobabooga's repository](https://github.com/oobabooga/text-generation-webui/tree/main/extensions/sd_api_pictures) to be STABLE, experimental developments and untested features are pushed in [Brawlence/SD_api_pics](https://github.com/Brawlence/SD_api_pics)
|
|
|
|
Lastest change:
|
|
1.1.0 → 1.1.1 Fixed not having Auto1111's metadata in received images
|
|
|
|
## Details
|
|
|
|
The image generation is triggered:
|
|
- manually through the 'Force the picture response' button while in `Manual` or `Immersive/Interactive` modes OR
|
|
- automatically in `Immersive/Interactive` mode if the words `'send|main|message|me'` are followed by `'image|pic|picture|photo|snap|snapshot|selfie|meme'` in the user's prompt
|
|
- always on in `Picturebook/Adventure` mode (if not currently suppressed by 'Suppress the picture response')
|
|
|
|
## Prerequisites
|
|
|
|
One needs an available instance of Automatic1111's webui running with an `--api` flag. Ain't tested with a notebook / cloud hosted one but should be possible.
|
|
To run it locally in parallel on the same machine, specify custom `--listen-port` for either Auto1111's or ooba's webUIs.
|
|
|
|
## Features overview
|
|
- Connection to API check (press enter in the address box)
|
|
- [VRAM management (model shuffling)](https://github.com/Brawlence/SD_api_pics/wiki/VRAM-management-feature)
|
|
- [Three different operation modes](https://github.com/Brawlence/SD_api_pics/wiki/Modes-of-operation) (manual, interactive, always-on)
|
|
- User-defined persistent settings via settings.json
|
|
|
|
### Connection check
|
|
|
|
Insert the Automatic1111's WebUI address and press Enter:
|
|
![API-check](https://raw.githubusercontent.com/Brawlence/SD_api_pics/main/illust/API-check.gif)
|
|
Green mark confirms the ability to communicate with Auto1111's API on this address. Red cross means something's not right (the ext won't work).
|
|
|
|
### Persistents settings
|
|
|
|
Create or modify the `settings.json` in the `text-generation-webui` root directory to override the defaults
|
|
present in script.py, ex:
|
|
|
|
```json
|
|
{
|
|
"sd_api_pictures-manage_VRAM": 1,
|
|
"sd_api_pictures-save_img": 1,
|
|
"sd_api_pictures-prompt_prefix": "(Masterpiece:1.1), detailed, intricate, colorful, (solo:1.1)",
|
|
"sd_api_pictures-sampler_name": "DPM++ 2M Karras"
|
|
}
|
|
```
|
|
|
|
will automatically set the `Manage VRAM` & `Keep original images` checkboxes and change the texts in `Prompt Prefix` and `Sampler name` on load.
|
|
|
|
---
|
|
|
|
## Demonstrations:
|
|
|
|
Those are examples of the version 1.0.0, but the core functionality is still the same
|
|
|
|
<details>
|
|
<summary>Conversation 1</summary>
|
|
|
|
![EXA1](https://user-images.githubusercontent.com/42910943/224866564-939a3bcb-e7cf-4ac0-a33f-b3047b55054d.jpg)
|
|
![EXA2](https://user-images.githubusercontent.com/42910943/224866566-38394054-1320-45cf-9515-afa76d9d7745.jpg)
|
|
![EXA3](https://user-images.githubusercontent.com/42910943/224866568-10ea47b7-0bac-4269-9ec9-22c387a13b59.jpg)
|
|
![EXA4](https://user-images.githubusercontent.com/42910943/224866569-326121ad-1ea1-4874-9f6b-4bca7930a263.jpg)
|
|
|
|
|
|
</details>
|
|
|
|
<details>
|
|
<summary>Conversation 2</summary>
|
|
|
|
![Hist1](https://user-images.githubusercontent.com/42910943/224865517-c6966b58-bc4d-4353-aab9-6eb97778d7bf.jpg)
|
|
![Hist2](https://user-images.githubusercontent.com/42910943/224865527-b2fe7c2e-0da5-4c2e-b705-42e233b07084.jpg)
|
|
![Hist3](https://user-images.githubusercontent.com/42910943/224865535-a38d94e7-8975-4a46-a655-1ae1de41f85d.jpg)
|
|
|
|
</details>
|
|
|