diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml index 17c51eb..167b29c 100644 --- a/.github/workflows/release.yml +++ b/.github/workflows/release.yml @@ -108,6 +108,7 @@ jobs: cd .. go install github.com/wailsapp/wails/v2/cmd/wails@latest rm -rf ./backend-python/wkv_cuda_utils + rm ./backend-python/rwkv_pip/beta/wkv_cuda.pyd rm ./backend-python/get-pip.py sed -i '1,2d' ./backend-golang/wsl_not_windows.go rm ./backend-golang/wsl.go @@ -140,6 +141,7 @@ jobs: cd .. go install github.com/wailsapp/wails/v2/cmd/wails@latest rm -rf ./backend-python/wkv_cuda_utils + rm ./backend-python/rwkv_pip/beta/wkv_cuda.pyd rm ./backend-python/get-pip.py sed -i '' '1,2d' ./backend-golang/wsl_not_windows.go rm ./backend-golang/wsl.go diff --git a/backend-python/rwkv_pip/beta/model.py b/backend-python/rwkv_pip/beta/model.py index 3cc0061..ca32e1a 100644 --- a/backend-python/rwkv_pip/beta/model.py +++ b/backend-python/rwkv_pip/beta/model.py @@ -94,6 +94,7 @@ if os.environ.get("RWKV_CUDA_ON") == "1": f"{current_path}/cuda/att_one_v5.cu", ], verbose=True, + extra_ldflags=["cublas.lib"], extra_cuda_cflags=[ "-t 4", "-std=c++17", diff --git a/backend-python/rwkv_pip/beta/wkv_cuda.pyd b/backend-python/rwkv_pip/beta/wkv_cuda.pyd new file mode 100644 index 0000000..ec10e31 Binary files /dev/null and b/backend-python/rwkv_pip/beta/wkv_cuda.pyd differ