Unverified Commit ba077e21 authored by DaniAndTheWeb's avatar DaniAndTheWeb Committed by GitHub
Browse files

Fix TORCH_COMMAND check

parent 2e172cf8

webui.sh

+1 −1
Original line number Diff line number Diff line
@@ -168,7 +168,7 @@ else 
    gpu_info=$(lspci | grep VGA)

    if echo "$gpu_info" | grep -q "AMD"

    then

        if [ -z ${TORCH_COMMAND+x} ]

        if [[ -z "${TORCH_COMMAND}" ]]

        then	    

            export TORCH_COMMAND="pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/rocm5.2"

        fi

CRA Git | Maintained and supported by SUSTech CRA and CCSE