Skip to content

Commit

Permalink
update defaultModelConfigs
Browse files Browse the repository at this point in the history
  • Loading branch information
@josStorer
josStorer committed Mar 2, 2024
1 parent 4f14074 commit 2947162
Showing 1 changed file with 18 additions and 18 deletions.
36 changes: 18 additions & 18 deletions frontend/src/pages/defaultConfigs.ts
View file
Original file line number Diff line number Diff line change
Expand Up @@ -207,7 +207,7 @@ export const defaultModelConfigsMac: ModelConfig[] = [
frequencyPenalty: 1
},
modelParameters: {
modelName: 'RWKV-5-World-1B5-v2-20231025-ctx4096.pth',
modelName: 'RWKV-x060-World-1B6-v2-20240208-ctx4096.pth',
device: 'WebGPU',
precision: 'nf4',
storedLayers: 41,
Expand All @@ -225,7 +225,7 @@ export const defaultModelConfigsMac: ModelConfig[] = [
frequencyPenalty: 1
},
modelParameters: {
modelName: 'RWKV-5-World-3B-v2-20231118-ctx16k.pth',
modelName: 'RWKV-x060-World-3B-v2-20240228-ctx4096.pth',
device: 'WebGPU',
precision: 'nf4',
storedLayers: 41,
Expand All @@ -243,7 +243,7 @@ export const defaultModelConfigsMac: ModelConfig[] = [
frequencyPenalty: 1
},
modelParameters: {
modelName: 'RWKV-4-World-CHNtuned-3B-v1-20230625-ctx4096.pth',
modelName: 'RWKV-x060-World-3B-v2-20240228-ctx4096.pth',
device: 'WebGPU',
precision: 'nf4',
storedLayers: 41,
Expand Down Expand Up @@ -333,7 +333,7 @@ export const defaultModelConfigsMac: ModelConfig[] = [
frequencyPenalty: 1
},
modelParameters: {
modelName: 'RWKV-5-World-1B5-v2-20231025-ctx4096.pth',
modelName: 'RWKV-x060-World-1B6-v2-20240208-ctx4096.pth',
device: 'MPS',
precision: 'fp32',
storedLayers: 41,
Expand All @@ -352,7 +352,7 @@ export const defaultModelConfigsMac: ModelConfig[] = [
frequencyPenalty: 1
},
modelParameters: {
modelName: 'RWKV-5-World-3B-v2-20231118-ctx16k.pth',
modelName: 'RWKV-x060-World-3B-v2-20240228-ctx4096.pth',
device: 'MPS',
precision: 'fp32',
storedLayers: 41,
Expand All @@ -371,7 +371,7 @@ export const defaultModelConfigsMac: ModelConfig[] = [
frequencyPenalty: 1
},
modelParameters: {
modelName: 'RWKV-4-World-CHNtuned-3B-v1-20230625-ctx4096.pth',
modelName: 'RWKV-x060-World-3B-v2-20240228-ctx4096.pth',
device: 'MPS',
precision: 'fp32',
storedLayers: 41,
Expand Down Expand Up @@ -412,7 +412,7 @@ export const defaultModelConfigs: ModelConfig[] = [
frequencyPenalty: 1
},
modelParameters: {
modelName: 'RWKV-5-World-1B5-v2-20231025-ctx4096.pth',
modelName: 'RWKV-x060-World-1B6-v2-20240208-ctx4096.pth',
device: 'CUDA',
precision: 'int8',
storedLayers: 41,
Expand All @@ -431,7 +431,7 @@ export const defaultModelConfigs: ModelConfig[] = [
frequencyPenalty: 1
},
modelParameters: {
modelName: 'RWKV-5-World-3B-v2-20231118-ctx16k.pth',
modelName: 'RWKV-x060-World-3B-v2-20240228-ctx4096.pth',
device: 'CUDA',
precision: 'int8',
storedLayers: 6,
Expand All @@ -450,7 +450,7 @@ export const defaultModelConfigs: ModelConfig[] = [
frequencyPenalty: 1
},
modelParameters: {
modelName: 'RWKV-5-World-1B5-v2-20231025-ctx4096.pth',
modelName: 'RWKV-x060-World-1B6-v2-20240208-ctx4096.pth',
device: 'CUDA',
precision: 'fp16',
storedLayers: 41,
Expand All @@ -469,7 +469,7 @@ export const defaultModelConfigs: ModelConfig[] = [
frequencyPenalty: 1
},
modelParameters: {
modelName: 'RWKV-5-World-3B-v2-20231118-ctx16k.pth',
modelName: 'RWKV-x060-World-3B-v2-20240228-ctx4096.pth',
device: 'CUDA',
precision: 'int8',
storedLayers: 24,
Expand All @@ -488,7 +488,7 @@ export const defaultModelConfigs: ModelConfig[] = [
frequencyPenalty: 1
},
modelParameters: {
modelName: 'RWKV-4-World-CHNtuned-3B-v1-20230625-ctx4096.pth',
modelName: 'RWKV-x060-World-3B-v2-20240228-ctx4096.pth',
device: 'CUDA',
precision: 'int8',
storedLayers: 24,
Expand Down Expand Up @@ -545,7 +545,7 @@ export const defaultModelConfigs: ModelConfig[] = [
frequencyPenalty: 1
},
modelParameters: {
modelName: 'RWKV-5-World-3B-v2-20231118-ctx16k.pth',
modelName: 'RWKV-x060-World-3B-v2-20240228-ctx4096.pth',
device: 'CUDA',
precision: 'int8',
storedLayers: 41,
Expand All @@ -564,7 +564,7 @@ export const defaultModelConfigs: ModelConfig[] = [
frequencyPenalty: 1
},
modelParameters: {
modelName: 'RWKV-4-World-CHNtuned-3B-v1-20230625-ctx4096.pth',
modelName: 'RWKV-x060-World-3B-v2-20240228-ctx4096.pth',
device: 'CUDA',
precision: 'int8',
storedLayers: 41,
Expand Down Expand Up @@ -621,7 +621,7 @@ export const defaultModelConfigs: ModelConfig[] = [
frequencyPenalty: 1
},
modelParameters: {
modelName: 'RWKV-5-World-3B-v2-20231118-ctx16k.pth',
modelName: 'RWKV-x060-World-3B-v2-20240228-ctx4096.pth',
device: 'CUDA',
precision: 'fp16',
storedLayers: 41,
Expand All @@ -640,7 +640,7 @@ export const defaultModelConfigs: ModelConfig[] = [
frequencyPenalty: 1
},
modelParameters: {
modelName: 'RWKV-4-World-CHNtuned-3B-v1-20230625-ctx4096.pth',
modelName: 'RWKV-x060-World-3B-v2-20240228-ctx4096.pth',
device: 'CUDA',
precision: 'fp16',
storedLayers: 41,
Expand Down Expand Up @@ -809,7 +809,7 @@ export const defaultModelConfigs: ModelConfig[] = [
frequencyPenalty: 1
},
modelParameters: {
modelName: 'RWKV-5-World-1B5-v2-20231025-ctx4096.pth',
modelName: 'RWKV-x060-World-1B6-v2-20240208-ctx4096.pth',
device: 'WebGPU',
precision: 'nf4',
storedLayers: 41,
Expand All @@ -827,7 +827,7 @@ export const defaultModelConfigs: ModelConfig[] = [
frequencyPenalty: 1
},
modelParameters: {
modelName: 'RWKV-5-World-3B-v2-20231118-ctx16k.pth',
modelName: 'RWKV-x060-World-3B-v2-20240228-ctx4096.pth',
device: 'WebGPU',
precision: 'nf4',
storedLayers: 41,
Expand All @@ -845,7 +845,7 @@ export const defaultModelConfigs: ModelConfig[] = [
frequencyPenalty: 1
},
modelParameters: {
modelName: 'RWKV-4-World-CHNtuned-3B-v1-20230625-ctx4096.pth',
modelName: 'RWKV-x060-World-3B-v2-20240228-ctx4096.pth',
device: 'WebGPU',
precision: 'nf4',
storedLayers: 41,
Expand Down

0 comments on commit 2947162

Please sign in to comment.