update defaultModelConfigs

This commit is contained in:
josc146
2024-02-03 20:39:18 +08:00
parent 26c2954c8e
commit 1c1c9e2c5f

View File

@@ -2,7 +2,7 @@ import { CompletionPreset } from '../types/completion';
import { ModelConfig } from '../types/configs';
export const defaultCompositionPrompt = '<pad>';
export const defaultCompositionABCPrompt='S:3\n' +
export const defaultCompositionABCPrompt = 'S:3\n' +
'B:9\n' +
'E:4\n' +
'B:9\n' +
@@ -12,7 +12,7 @@ export const defaultCompositionABCPrompt='S:3\n' +
'L:1/8\n' +
'M:3/4\n' +
'K:D\n' +
' Bc |"G" d2 cB"A" A2 FE |"Bm" F2 B4 F^G |'
' Bc |"G" d2 cB"A" A2 FE |"Bm" F2 B4 F^G |';
export const defaultPresets: CompletionPreset[] = [{
name: 'Writer',
@@ -186,7 +186,7 @@ export const defaultModelConfigsMac: ModelConfig[] = [
frequencyPenalty: 1
},
modelParameters: {
modelName: 'RWKV-4-World-7B-v1-20230626-ctx4096.pth',
modelName: 'RWKV-5-World-7B-v2-20240128-ctx4096.pth',
device: 'WebGPU',
precision: 'nf4',
storedLayers: 41,
@@ -204,7 +204,7 @@ export const defaultModelConfigsMac: ModelConfig[] = [
frequencyPenalty: 1
},
modelParameters: {
modelName: 'RWKV-4-World-CHNtuned-7B-v1-20230709-ctx4096.pth',
modelName: 'RWKV-5-World-7B-v2-20240128-ctx4096.pth',
device: 'WebGPU',
precision: 'nf4',
storedLayers: 41,
@@ -315,7 +315,7 @@ export const defaultModelConfigsMac: ModelConfig[] = [
frequencyPenalty: 1
},
modelParameters: {
modelName: 'RWKV-4-World-7B-v1-20230626-ctx4096.pth',
modelName: 'RWKV-5-World-7B-v2-20240128-ctx4096.pth',
device: 'MPS',
precision: 'fp32',
storedLayers: 41,
@@ -432,7 +432,7 @@ export const defaultModelConfigs: ModelConfig[] = [
frequencyPenalty: 1
},
modelParameters: {
modelName: 'RWKV-4-World-7B-v1-20230626-ctx4096.pth',
modelName: 'RWKV-5-World-7B-v2-20240128-ctx4096.pth',
device: 'CUDA',
precision: 'int8',
storedLayers: 8,
@@ -451,7 +451,7 @@ export const defaultModelConfigs: ModelConfig[] = [
frequencyPenalty: 1
},
modelParameters: {
modelName: 'RWKV-4-World-CHNtuned-7B-v1-20230709-ctx4096.pth',
modelName: 'RWKV-5-World-7B-v2-20240128-ctx4096.pth',
device: 'CUDA',
precision: 'int8',
storedLayers: 8,
@@ -508,7 +508,7 @@ export const defaultModelConfigs: ModelConfig[] = [
frequencyPenalty: 1
},
modelParameters: {
modelName: 'RWKV-4-World-7B-v1-20230626-ctx4096.pth',
modelName: 'RWKV-5-World-7B-v2-20240128-ctx4096.pth',
device: 'CUDA',
precision: 'int8',
storedLayers: 18,
@@ -527,7 +527,7 @@ export const defaultModelConfigs: ModelConfig[] = [
frequencyPenalty: 1
},
modelParameters: {
modelName: 'RWKV-4-World-CHNtuned-7B-v1-20230709-ctx4096.pth',
modelName: 'RWKV-5-World-7B-v2-20240128-ctx4096.pth',
device: 'CUDA',
precision: 'int8',
storedLayers: 18,
@@ -584,7 +584,7 @@ export const defaultModelConfigs: ModelConfig[] = [
frequencyPenalty: 1
},
modelParameters: {
modelName: 'RWKV-4-World-7B-v1-20230626-ctx4096.pth',
modelName: 'RWKV-5-World-7B-v2-20240128-ctx4096.pth',
device: 'CUDA',
precision: 'int8',
storedLayers: 27,
@@ -603,7 +603,7 @@ export const defaultModelConfigs: ModelConfig[] = [
frequencyPenalty: 1
},
modelParameters: {
modelName: 'RWKV-4-World-CHNtuned-7B-v1-20230709-ctx4096.pth',
modelName: 'RWKV-5-World-7B-v2-20240128-ctx4096.pth',
device: 'CUDA',
precision: 'int8',
storedLayers: 27,
@@ -622,7 +622,7 @@ export const defaultModelConfigs: ModelConfig[] = [
frequencyPenalty: 1
},
modelParameters: {
modelName: 'RWKV-4-World-7B-v1-20230626-ctx4096.pth',
modelName: 'RWKV-5-World-7B-v2-20240128-ctx4096.pth',
device: 'CUDA',
precision: 'int8',
storedLayers: 41,
@@ -641,7 +641,7 @@ export const defaultModelConfigs: ModelConfig[] = [
frequencyPenalty: 1
},
modelParameters: {
modelName: 'RWKV-4-World-CHNtuned-7B-v1-20230709-ctx4096.pth',
modelName: 'RWKV-5-World-7B-v2-20240128-ctx4096.pth',
device: 'CUDA',
precision: 'int8',
storedLayers: 41,
@@ -660,7 +660,7 @@ export const defaultModelConfigs: ModelConfig[] = [
frequencyPenalty: 1
},
modelParameters: {
modelName: 'RWKV-4-World-7B-v1-20230626-ctx4096.pth',
modelName: 'RWKV-5-World-7B-v2-20240128-ctx4096.pth',
device: 'CUDA',
precision: 'fp16',
storedLayers: 41,
@@ -679,7 +679,7 @@ export const defaultModelConfigs: ModelConfig[] = [
frequencyPenalty: 1
},
modelParameters: {
modelName: 'RWKV-4-World-CHNtuned-7B-v1-20230709-ctx4096.pth',
modelName: 'RWKV-5-World-7B-v2-20240128-ctx4096.pth',
device: 'CUDA',
precision: 'fp16',
storedLayers: 41,
@@ -788,7 +788,7 @@ export const defaultModelConfigs: ModelConfig[] = [
frequencyPenalty: 1
},
modelParameters: {
modelName: 'RWKV-4-World-7B-v1-20230626-ctx4096.pth',
modelName: 'RWKV-5-World-7B-v2-20240128-ctx4096.pth',
device: 'WebGPU',
precision: 'nf4',
storedLayers: 41,
@@ -806,7 +806,7 @@ export const defaultModelConfigs: ModelConfig[] = [
frequencyPenalty: 1
},
modelParameters: {
modelName: 'RWKV-4-World-CHNtuned-7B-v1-20230709-ctx4096.pth',
modelName: 'RWKV-5-World-7B-v2-20240128-ctx4096.pth',
device: 'WebGPU',
precision: 'nf4',
storedLayers: 41,