[Modify]fix bug of install

LISTENAI · Aug 14, 2023 · 23911ce · 23911ce
1 parent 0d147a8
commit 23911ce
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/linger/__init__.py b/linger/__init__.py
@@ -1,3 +1,9 @@
+import os
+import sys
+
+cur_dir = os.path.abspath(os.path.dirname(__file__))
+sys.path.append(os.path.join(cur_dir, "lib"))
+
 from .__version import __version__, version_info
 from .config import *
 from .conv_bn_fuser import EmptyBatchNorm, FuseBNIntoConv, FuseConvBNAheadRelu