From e40549db3e167b53d8a8101f032f8c1ef2de4684 Mon Sep 17 00:00:00 2001 From: rejoicesyc Date: Tue, 6 Jun 2023 09:14:54 +0000 Subject: [PATCH] fix converter header files including path --- converter/convert_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/converter/convert_model.py b/converter/convert_model.py index bf89561..2b0fafe 100755 --- a/converter/convert_model.py +++ b/converter/convert_model.py @@ -198,7 +198,7 @@ def convert_model(path: str): name="wkv_cuda_export", sources=[os.path.join(current_path, "cpp_save_tensor.cpp")], # add ../include to include path - extra_include_paths=[os.path.join(current_path, "../include/rwkv/rwkv")], + extra_include_paths=[os.path.join(current_path, "../include/")], ) w = torch.load(path, map_location="cpu")