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

improve error messages

Browse Source
This commit is contained in:
josc146
2023-06-23 13:55:45 +08:00
parent 5c9b4a4c05
commit 447f4572b1
4 changed files with 21 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
frontend/src/components/RunButton.tsx
View File

@@ -207,11 +207,21 @@ export const RunButton: FC<{ onClickRun?: MouseEventHandler, iconMode?: boolean
toast(t('Loading Model'), { type: 'info' });
} else {
commonStore.setStatus({ status: ModelStatus.Offline });
toast(t('Failed to switch model') + ' - ' + await r.text(), { type: 'error' });
const error = await r.text();
const errorsMap = {
'not enough memory': 'Memory is not enough, try to increase the virtual memory or use a smaller model.',
'not compiled with CUDA': 'Bad pytorch version, please reinstall pytorch with cuda.',
'invalid header or archive is corrupted': 'The model file is corrupted, please download again.',
'no NVIDIA driver': 'Found no NVIDIA driver, please install the latest driver.',
'CUDA out of memory': 'VRAM is not enough, please reduce stored layers or use a lower precision in Configs page.'
};
const matchedError = Object.entries(errorsMap).find(([key, _]) => error.includes(key));
const message = matchedError ? t(matchedError[1]) : error;
toast(t('Failed to switch model') + ' - ' + message, { autoClose: 5000, type: 'error' });
}
}).catch((e) => {
commonStore.setStatus({ status: ModelStatus.Offline });
toast(t('Failed to switch model') + ' - ' + e.message || e, { type: 'error' });
toast(t('Failed to switch model') + ' - ' + (e.message || e), { type: 'error' });
});
}
}).catch(() => {