From c853c5b60bca4223627717eb40661b833e2ca2de Mon Sep 17 00:00:00 2001 From: josc146 Date: Wed, 6 Dec 2023 23:09:39 +0800 Subject: [PATCH] chore --- backend-python/requirements.txt | 2 +- backend-python/requirements_without_cyac.txt | 2 +- frontend/src/components/RunButton.tsx | 4 ++-- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/backend-python/requirements.txt b/backend-python/requirements.txt index 48ae79c..54e5ee6 100644 --- a/backend-python/requirements.txt +++ b/backend-python/requirements.txt @@ -1,7 +1,7 @@ torch torchvision torchaudio -rwkv==0.8.16 +rwkv==0.8.22 langchain==0.0.322 fastapi==0.104.0 uvicorn==0.23.2 diff --git a/backend-python/requirements_without_cyac.txt b/backend-python/requirements_without_cyac.txt index 618d6ff..f4a9f05 100644 --- a/backend-python/requirements_without_cyac.txt +++ b/backend-python/requirements_without_cyac.txt @@ -1,7 +1,7 @@ torch torchvision torchaudio -rwkv==0.8.16 +rwkv==0.8.22 langchain==0.0.322 fastapi==0.104.0 uvicorn==0.23.2 diff --git a/frontend/src/components/RunButton.tsx b/frontend/src/components/RunButton.tsx index f8dbd9d..e330c62 100644 --- a/frontend/src/components/RunButton.tsx +++ b/frontend/src/components/RunButton.tsx @@ -169,7 +169,7 @@ export const RunButton: FC<{ onClickRun?: MouseEventHandler, iconMode?: boolean }); } commonStore.setStatus({ status: ModelStatus.Loading }); - const loadingId = toast(t('Loading Model'), { type: 'info' }); + const loadingId = toast(t('Loading Model'), { type: 'info', autoClose: false }); if (!webgpu) { updateConfig({ max_tokens: modelConfig.apiParameters.maxResponseToken, @@ -232,7 +232,7 @@ export const RunButton: FC<{ onClickRun?: MouseEventHandler, iconMode?: boolean if ((modelConfig.modelParameters.device === 'CUDA' || modelConfig.modelParameters.device === 'CUDA-Beta') && modelConfig.modelParameters.storedLayers < modelConfig.modelParameters.maxStoredLayers && commonStore.monitorData && commonStore.monitorData.totalVram !== 0 && - (commonStore.monitorData.usedVram / commonStore.monitorData.totalVram) < 0.85) + (commonStore.monitorData.usedVram / commonStore.monitorData.totalVram) < 0.9) toast(t('You can increase the number of stored layers in Configs page to improve performance'), { type: 'info' }); toastWithButton(t('Startup Completed'), t(buttonName), buttonFn, { type: 'success', autoClose: 3000 }); } else if (r.status === 304) {