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

exact avoidOverflow

Browse Source
This commit is contained in:
josc146 2023-06-21 23:08:34 +08:00
parent 21910af96a
commit 82a68593bb
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
frontend/src/utils/index.tsx
View File

@ -126,7 +126,9 @@ export const getStrategy = (modelConfig: ModelConfig | undefined = undefined) =>
let params: ModelParameters;
if (modelConfig) params = modelConfig.modelParameters;
else params = commonStore.getCurrentModelConfig().modelParameters;
const avoidOverflow = params.modelName.toLowerCase().includes('world') && params.precision !== 'fp32';
const modelName = params.modelName.toLowerCase();
const avoidOverflow = params.precision !== 'fp32' && modelName.includes('world') && (modelName.includes('0.1b') || modelName.includes('0.4b') ||
modelName.includes('1.5b') || modelName.includes('1b5'));
let strategy = '';
switch (params.device) {
case 'CPU':