Skip to content

Instantly share code, notes, and snippets.

@racinmat
Last active June 2, 2023 09:03
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save racinmat/74d0f08618c1c39edf1e06a0bddd04ce to your computer and use it in GitHub Desktop.
Save racinmat/74d0f08618c1c39edf1e06a0bddd04ce to your computer and use it in GitHub Desktop.
Jak si zprovoznit AI generátor obrázků doma, a technické informace

Návod na zprovoznění AI generátoru obrázků doma a technické informace

Demo běželo na běžné herní GPU, konkrétně GTX 1080. Software by měl jít bez problémů nainstalovat na jakoukoli Nvidia GPU s dostatečnou pamětí (aspoň 4GB, spíš 6GB, u 4GB povypínejte ostatní věci). Lze to zprovoznit i na AMD GPU, ale je to komplikovanější.

Generátor obrázků je neuronová síť založená na Stable Diffusion v1.5. Grafické rozhranní (klikátko je web UI od Automatic1111. Pro účely AnimeFestu jsem rozhraní upravil, a všechny úpravy jsou v mém forku: https://github.com/racinmat/stable-diffusion-webui Tento fork primárně umožňuje schovat většinu UI, aby to bylo jednodušší pro návštěvníky. Pokud si chcete generovat doma a hrát si s nastavením, doporučuji původní repo: https://github.com/AUTOMATIC1111/stable-diffusion-webui.

Pokud chcete použít můj fork, postupujte podle návodu na instalaci z https://github.com/AUTOMATIC1111/stable-diffusion-webui#installation-and-running, ale místo git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git použijte git clone https://github.com/racinmat/stable-diffusion-webui.git

Pokud chcete použít můj fork, ale mít přístupné i jiné taby (např. Settings apod.), upravte řádek https://github.com/racinmat/stable-diffusion-webui/blob/master/config-af.json#L115 a odeberte všechny taby, které chcete vidět.

Použitý model: Kenshi 01, pruned, f16

Použitý variační autoencoder (VAE): kl-f8-anime2

Návod na spuštění a technické požadavky: viz readme v repu https://github.com/AUTOMATIC1111/stable-diffusion-webui

Jeden z mnoha návodů na zprovoznění: https://stable-diffusion-art.com/install-windows/

AUTOMATIC1111/stable-diffusion-webui má poměrně rozsáhlou wiki se spoustou návodů: https://github.com/AUTOMATIC1111/stable-diffusion-webui/wiki

Do stable diffusion web ui existuje mnoho extensions, za zmínku stojí např. kontrola pózy na obrázku https://github.com/AUTOMATIC1111/stable-diffusion-webui/wiki/Extensions#openpose-editor nebo např. výše zmíněný diffusion defender https://github.com/AUTOMATIC1111/stable-diffusion-webui/wiki/Extensions#diffusion-defender

Přednáška o AI a zdroje na letošním AF:

Modifikace mého forku

Oproti původnímu repu je u mnoha tlačítek přidán label, aby byly v konfiguráku. U většiny UI prvků je nastaveno, aby byly skryté, abychom měli jednodušší UI pro běžné uživatele. Původní repo má plné UI. Puštění mého forku s původními konfiguráky je možné po nastavení:

set COMMANDLINE_ARGS=--theme dark --ui-settings-file=config.json --ui-config-file=ui-config-default.json

v webui-user.bat.

Většina skrytých parametrů je nastavená na rozumné defaultní hodnoty.

Je tam doinstalována extensiona https://github.com/WildBanjos/DiffusionDefender, konkrétně můj fork https://github.com/racinmat/DiffusionDefender, s vlastním blacklistem, aby během festivalo nebylo snadné generovat porno. Je tam také schován a zahardcoděn negative prompt komplikující generování porna. Není problém doma pustit stable diffusion s jiným negative promptem a nebo extension/s prázdným blacklistem v extension, a generovat si doma porno.

modely

Web UI umožňuje používat mnoho modelů a přepínat mezi nimi, modely můžete najít na těchto webech, kde jsou zdarma volně dostupné:

odkud stahnete model, dáte do příslušné složky, a ve Web UI půjde načíst.

poznámky

Při spuštění se stahuje pytorch_model.bin, co má cca 1.7GB, viz AUTOMATIC1111/stable-diffusion-webui#9182, dá se stahnout předem z https://huggingface.co/openai/clip-vit-large-patch14/tree/main

odkazy a materiály

language models:

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment