gemma2 / model.yml
thuannh's picture
Update model.yml
1e5e4a0 verified
raw
history blame
531 Bytes
name: gemma2
model: gemma2:2B
version: 1
# Results Preferences
stop:
- <end_of_turn>
- <eos>
top_p: 0.95
temperature: 0.7
frequency_penalty: 0
presence_penalty: 0
max_tokens: 4096 # Infer from base config.json -> max_position_embeddings
stream: true # true | false
# Engine / Model Settings
ngl: 300 # Infer from base config.json -> num_attention_heads
ctx_len: 4096 # Infer from base config.json -> max_position_embeddings
engine: llama-cpp
prompt_template: "<start_of_turn>user\n{prompt}<end_of_turn>\n<start_of_turn>model"