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

MX, Tesla P, Quadro P, NVIDIA P, TITAN X, RTX A series, TITAN RTX and RTX TITAN Ada

Browse Source
This commit is contained in:
josc146 2023-06-03 12:46:56 +08:00
parent b01584c49e
commit f2ec1067bf
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -309,9 +309,9 @@ export function toastWithButton(text: string, buttonText: string, onClickButton:
} }
export function getSupportedCustomCudaFile() { export function getSupportedCustomCudaFile() {
if ([' 10', ' 16', ' 20', ' 30', 'P40', 'P104', 'P106'].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'].some(v => commonStore.status.device_name.includes(v)))
return './backend-python/wkv_cuda_utils/wkv_cuda10_30.pyd'; return './backend-python/wkv_cuda_utils/wkv_cuda10_30.pyd';
else if ([' 40'].some(v => commonStore.status.device_name.includes(v))) else if ([' 40', 'RTX TITAN Ada'].some(v => commonStore.status.device_name.includes(v)))
return './backend-python/wkv_cuda_utils/wkv_cuda40.pyd'; return './backend-python/wkv_cuda_utils/wkv_cuda40.pyd';
else else
return ''; return '';