Add more params for llama
Signed-off-by: baalajimaestro <me@baalajimaestro.me>
This commit is contained in:
parent
55ce7d9fc2
commit
110deb68cf
1 changed files with 14 additions and 3 deletions
17
run_model.sh
17
run_model.sh
|
@ -5,9 +5,20 @@
|
|||
sleep 10
|
||||
|
||||
if [ -n "$INIT_MODELS" ]; then
|
||||
/ollama/ollama rm model || true
|
||||
echo "FROM /models/$MODEL_NAME" > /tmp/Modelfile
|
||||
/ollama/ollama create -f /tmp/Modelfile model
|
||||
/ollama/ollama rm model || true
|
||||
echo "FROM /models/$MODEL_NAME" > /tmp/Modelfile
|
||||
echo "PARAMETER temperature 1" >> /tmp/Modelfile
|
||||
echo "PARAMETER num_ctx 4096" >> /tmp/Modelfile
|
||||
echo 'PARAMETER stop ["<|im_start|>","<|im_end|>"]' >> /tmp/Modelfile
|
||||
echo 'TEMPLATE """{{ if .System }}<|im_start|>system' >> /tmp/Modelfile
|
||||
echo "{{ .System }}<|im_end|>" >> /tmp/Modelfile
|
||||
echo "{{ end }}{{ if .Prompt }}<|im_start|>user" >> /tmp/Modelfile
|
||||
echo "{{ .Prompt }}<|im_end|>" >> /tmp/Modelfile
|
||||
echo "{{ end }}<|im_start|>assistant" >> /tmp/Modelfile
|
||||
echo "{{ .Response }}<|im_end|>" >> /tmp/Modelfile
|
||||
echo '"""' >> /tmp/Modelfile
|
||||
echo 'SYSTEM """You are Samantha, a sentient AI."""' >> /tmp/Modelfile
|
||||
/ollama/ollama create -f /tmp/Modelfile model
|
||||
fi
|
||||
|
||||
/ollama/ollama run model
|
||||
|
|
Loading…
Reference in a new issue