ai-suite-rocm-local/stablediffusion-rocm/run.sh
2024-06-30 22:06:23 +02:00

19 lines
346 B
Bash
Executable File

#!/bin/bash
source ../utils.sh
python_exec=venv/bin/python3.10
# Main function
main() {
# Create virtual environment
use_venv
# Prints ROCM info with available GPUs
rocm-smi
# Start SD
TORCH_BLAS_PREFER_HIPBLASLT=0 $python_exec webui/launch.py --listen --enable-insecure-extension-access --opt-split-attention
}
main