Spaces:
Running
Running
File size: 515 Bytes
47c0b4f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
// Contains information about pre-set models
type ModelInfo = {
name: string
params: number
hidden_size: number
num_hidden_layers: number
}
export const MODELS: ModelInfo[] = [
{
name: 'LLaMA 2 (7B)',
params: 7,
hidden_size: 4096,
num_hidden_layers: 32,
},
{
name: 'LLaMA 2 (13B)',
params: 13,
hidden_size: 5120,
num_hidden_layers: 40,
},
{
name: 'LLaMA 2 (70B)',
params: 70,
hidden_size: 8192,
num_hidden_layers: 80,
},
]
export default MODELS
|