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

support nvidia 16xx

Browse Source
This commit is contained in:
josc146 2023-05-24 21:27:48 +08:00
parent c7ed4b07c2
commit 1676c5b7e6
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -283,7 +283,7 @@ export function toastWithButton(text: string, buttonText: string, onClickButton:
} }
export function getSupportedCustomCudaFile() { export function getSupportedCustomCudaFile() {
if ([' 10', ' 20', ' 30'].some(v => commonStore.status.device_name.includes(v))) if ([' 10', ' 16', ' 20', ' 30'].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'].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';