Skip to content

Mateusz-Dera/ROCm-AI-Installer

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 

Repository files navigation

ROCm-AI-Installer

A script that automatically installs all the required stuff to run selected AI interfaces on AMD Radeon 7900XTX.

Info

Version

Part of the installation script is based on this guide: https://github.com/nktice/AMD-AI/blob/main/ROCm6.0.md

Text generation

Name Enviroment Links Additional information
KoboldCPP Python 3.11 venv https://github.com/YellowRoseCx/koboldcpp-rocm
Text generation web UI Python 3.11 venv https://github.com/oobabooga/text-generation-webui
https://github.com/arlo-phoenix/bitsandbytes-rocm-5.6
https://github.com/ROCmSoftwarePlatform/flash-attention
https://github.com/turboderp/exllamav2
1. Tested: ExLlamav2, Transformers, llama.ccp
2. Smart Context and Silero TTS Requrements for Superbooga are installed, but the extension is not enabled by default
SillyTavern (1.12.0-preview)
SillyTavern-Extras
Node
Python 3.11 venv
https://github.com/SillyTavern/SillyTavern
https://github.com/SillyTavern/SillyTavern-Extras
1. SillyTavern and SillyTavern-Extras are launched separately
2. SillyTavern must be connected to SillyTavern-Extras in settings
3. Smart Context and Silero TTS are installed and enabled by default
4. Smart Context requires an additional extension download in settings
5. Smart Context and Silero TTS extensions must be manually configured in SillyTavern settings

Image generation

Name Enviroment Links Additional information
Stable Diffusion web UI Python 3.10 venv https://github.com/AUTOMATIC1111/stable-diffusion-webui 1. Startup parameters are in the webui-user.sh file
ANIMAGINE XL 3.1 Python 3.11 venv https://huggingface.co/spaces/cagliostrolab/animagine-xl-3.1
https://huggingface.co/cagliostrolab/animagine-xl-3.1

Music generation

Name Enviroment Links Additional information
AudioCraft Python 3.10 venv https://github.com/facebookresearch/audiocraft

Voice generation

Name Enviroment Links Additional information
WhisperSpeech web UI Python 3.11 venv https://github.com/Mateusz-Dera/whisperspeech-webui
https://github.com/collabora/WhisperSpeech
https://github.com/ROCmSoftwarePlatform/flash-attention

3D generation

Name Enviroment Links Additional information
TripoSR Python3.11 venv https://github.com/VAST-AI-Research/TripoSR
https://github.com/ROCmSoftwarePlatform/flash-attention.git
1. It uses PyTorch ROCm, but torchmcubes is built for the CPU. This method is still faster than using just PyTorch CPU-only version.

Test platform:

Name Info
CPU AMD Ryzen 7900X3D (iGPU disabled in BIOS)
GPU AMD Radeon 7900XTX
RAM 64GB DDR5 6600MHz
Motherboard ASRock B650E PG Riptide WiFi (2.10)
OS Ubuntu 22.04
Kernel 6.5.0-28-generic
ROCm 6.1

Instalation:

Note

First startup after installation of the selected interface may take longer.

Important

This script does not download any models. If the interface does not have defaults, download your own.

Caution

If you update, back up your settings and models. Reinstallation deletes the previous directories.

  1. Add the user to the required groups.
sudo adduser `whoami` video
sudo adduser `whoami` render
  1. Reboot
sudo reboot
  1. Run installer
bash ./install.sh
  1. Select installation path.
  2. Select ROCm installation if you are upgrading or running the script for the first time.
  3. Install selected interfaces
  4. Go to the installation path with the selected interface and run:
./run.sh