From 1465908574cc6a66a467773b46fbc228d553822a Mon Sep 17 00:00:00 2001 From: josc146 Date: Wed, 21 Jun 2023 13:48:09 +0800 Subject: [PATCH] update SupportedCustomCuda --- frontend/src/utils/index.tsx | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/frontend/src/utils/index.tsx b/frontend/src/utils/index.tsx index 2b9733f..8294e52 100644 --- a/frontend/src/utils/index.tsx +++ b/frontend/src/utils/index.tsx @@ -318,9 +318,10 @@ export function toastWithButton(text: string, buttonText: string, onClickButton: } export function getSupportedCustomCudaFile() { - if ([' 10', ' 16', ' 20', ' 30', 'MX', 'Tesla P', 'Quadro P', 'NVIDIA P', 'TITAN X', 'TITAN RTX', 'RTX A'].some(v => commonStore.status.device_name.includes(v))) + if ([' 10', ' 16', ' 20', ' 30', 'MX', 'Tesla P', 'Quadro P', 'NVIDIA P', 'TITAN X', 'TITAN RTX', 'RTX A', + 'Quadro RTX 4000', 'Quadro RTX 5000', 'Tesla T4', 'NVIDIA A10', 'NVIDIA A40'].some(v => commonStore.status.device_name.includes(v))) return './backend-python/wkv_cuda_utils/wkv_cuda10_30.pyd'; - else if ([' 40', 'RTX TITAN Ada'].some(v => commonStore.status.device_name.includes(v))) + else if ([' 40', 'RTX 5000 Ada', 'RTX 6000 Ada', 'RTX TITAN Ada', 'NVIDIA L40'].some(v => commonStore.status.device_name.includes(v))) return './backend-python/wkv_cuda_utils/wkv_cuda40.pyd'; else return '';