update run script to incorporate GPU layers

This commit is contained in:
James Ravenscroft 2023-08-26 16:03:16 +01:00
parent 326e76c9bb
commit 88683abe50

5
run.sh
View File

@ -1,3 +1,6 @@
#!/bin/sh
if [ -z "$GPU_LAYERS" ]; then
/app/turbopilot -t $THREADS -m $MODEL_TYPE -f $MODEL
else
/app/turbopilot -t $THREADS -m $MODEL_TYPE -f $MODEL --ngl $GPU_LAYERS
fi