### for ROCm | |
git clone https://github.com/kohya-ss/sd-scripts.git | |
cd sd-scripts | |
git clone https://huggingface.co/aka7774/frog_bench.git | |
sudo apt install python3.10-venv | |
python3 -m venv venv | |
venv/bin/pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm5.4.2 | |
venv/bin/pip3 install --upgrade -r requirements.txt | |
sudo apt-get install -y libgl1-mesa-dev | |
venv/bin/pip3 uninstall tensorflow | |
venv/bin/pip3 install tensorflow-rocm | |