2024-08-19 18:34:52 +02:00
|
|
|
#!/bin/bash
|
|
|
|
|
source ../utils.sh
|
2024-08-27 15:37:35 +02:00
|
|
|
python_exec="$(pwd)/venv/bin/python3.10"
|
2024-08-19 18:34:52 +02:00
|
|
|
|
|
|
|
|
# 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
|
