diff --git a/backend-python/rwkv_pip/beta/model.py b/backend-python/rwkv_pip/beta/model.py index ca32e1a..1c90fbf 100644 --- a/backend-python/rwkv_pip/beta/model.py +++ b/backend-python/rwkv_pip/beta/model.py @@ -94,7 +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_ldflags=["cublas.lib" if os.name == "nt" else ""], extra_cuda_cflags=[ "-t 4", "-std=c++17",