Spaces:
Running
Running
""" | |
#!/bin/bash | |
CUDA_ENABLED=${CUDA_ENABLED:-true} | |
DEVICE="" | |
if [ "${CUDA_ENABLED}" != "true" ]; then | |
DEVICE="--device cpu" | |
fi | |
exec python tools/webui.py ${DEVICE} | |
""" | |
import os | |
import torch | |
if torch.cuda.is_available(): | |
DEVICE = "cuda" | |
else: | |
DEVICE = "cpu" | |
DEVICE = f"--device {DEVICE}" | |
os.system(f"python tools/webui.py {DEVICE}") |