Merge pull request #32 from rejoicesyc/main

fix converter header files including path
harrisonvanderbyl · Jun 6, 2023 · e851a67 · e851a67
2 parents 864acf3 + e40549d
commit e851a67
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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")