diff --git a/frontend/src/utils/index.tsx b/frontend/src/utils/index.tsx index dceb258..d512d3a 100644 --- a/frontend/src/utils/index.tsx +++ b/frontend/src/utils/index.tsx @@ -283,7 +283,7 @@ export function toastWithButton(text: string, buttonText: string, onClickButton: } 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'; else if ([' 40'].some(v => commonStore.status.device_name.includes(v))) return './backend-python/wkv_cuda_utils/wkv_cuda40.pyd';