custom strategy mode

2023-05-31 12:26:10 +08:00
parent 8291c50058
commit 9f5d15a7d5
4 changed files with 100 additions and 62 deletions
--- a/frontend/src/components/RunButton.tsx
+++ b/frontend/src/components/RunButton.tsx
@@ -136,7 +136,7 @@ export const RunButton: FC<{ onClickRun?: MouseEventHandler, iconMode?: boolean
            });

            let customCudaFile = '';
-            if (modelConfig.modelParameters.useCustomCuda) {
+            if (modelConfig.modelParameters.device != 'CPU' && modelConfig.modelParameters.useCustomCuda) {
              customCudaFile = getSupportedCustomCudaFile();
              if (customCudaFile) {
                FileExists('./py310/Lib/site-packages/rwkv/model.py').then((exist) => {