File tree Expand file tree Collapse file tree 1 file changed +6
-8
lines changed Expand file tree Collapse file tree 1 file changed +6
-8
lines changed Original file line number Diff line number Diff line change 11#! /bin/bash
22
3- if ! command -v nvidia-smi & > /dev/null; then
3+ if ! command -v nvidia-smi & > /dev/null; then
44 echo " Error: 'nvidia-smi' command not found."
55 exit 1
66fi
77
88compute_cap=$( nvidia-smi --query-gpu=compute_cap --format=csv | sed -n ' 2p' | sed ' s/\.//g' )
99
10- if [ ${compute_cap} -eq 75 ]
11- then
10+ if [ ${compute_cap} -eq 75 ]; then
1211 exec text-embeddings-router-75 " $@ "
13- elif [ ${compute_cap} -ge 80 -a ${compute_cap} -lt 90 ]
14- then
12+ elif [ ${compute_cap} -ge 80 -a ${compute_cap} -lt 90 ]; then
1513 exec text-embeddings-router-80 " $@ "
16- elif [ ${compute_cap} -eq 90 ]
17- then
14+ elif [ ${compute_cap} -eq 90 ]; then
1815 exec text-embeddings-router-90 " $@ "
1916else
20- echo " cuda compute cap ${compute_cap} is not supported" ; exit 1
17+ echo " cuda compute cap ${compute_cap} is not supported"
18+ exit 1
2119fi
You can’t perform that action at this time.
0 commit comments