theluyuan/RWKV-Runner
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update defaultModelConfigs

Browse Source
This commit is contained in:
josc146 2024-02-03 20:39:18 +08:00
parent 26c2954c8e
commit 1c1c9e2c5f
1 changed files with 17 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
frontend/src/pages/defaultConfigs.ts
View File

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