Skip to content

Commit

Permalink
feat: added Gemma 2 prompt template
Browse files Browse the repository at this point in the history
  • Loading branch information
Vali-98 committed Jul 31, 2024
1 parent 24ba821 commit 027d9ab
Showing 1 changed file with 21 additions and 1 deletion.
22 changes: 21 additions & 1 deletion app/constants/Instructs.ts
Original file line number Diff line number Diff line change
Expand Up @@ -6,9 +6,9 @@ import { createJSONStorage, persist } from 'zustand/middleware'

import { Global } from './GlobalValues'
import { Logger } from './Logger'
import { mmkv, mmkvStorage } from './MMKV'
import { Tokenizer } from './Tokenizer'
import { replaceMacros } from './Utils'
import { mmkv, mmkvStorage } from './MMKV'

const defaultInstructs: InstructType[] = [
{
Expand Down Expand Up @@ -111,6 +111,26 @@ const defaultInstructs: InstructType[] = [
examples: true,
format_type: 0,
},
{
system_prompt: "Write {{char}}'s next reply in a chat between {{char}} and {{user}}.",
system_prefix: '<|start_of_turn|>user\n',
system_suffix: '<|end_of_turn|>\n',
input_prefix: '<|start_of_turn|>user\n',
input_suffix: '<|end_of_turn|>\n',
output_prefix: '<|start_of_turn|>model',
output_suffix: '<|end_of_turn|>\n',
stop_sequence: '<|end_of_turn|>\n',
user_alignment_message: '',
activation_regex: '',
wrap: false,
macro: false,
names: false,
names_force_groups: false,
name: 'Gemma 2',
timestamp: false,
examples: true,
format_type: 0,
},
]

type InstructState = {
Expand Down

0 comments on commit 027d9ab

Please sign in to comment.