Merge pull request #359 from purpleroc/master

add support macos
ztxz16 · Nov 8, 2023 · f97fff8 · f97fff8
2 parents 96159de + d749e38
commit f97fff8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/tools/fastllm_pytools/llm.py b/tools/fastllm_pytools/llm.py
@@ -7,6 +7,8 @@
 import platform
 if platform.system() == 'Windows':
     fastllm_lib = ctypes.cdll.LoadLibrary(os.path.join(os.path.split(os.path.realpath(__file__))[0], "fastllm_tools.dll"))
+elif platform.system() == 'Darwin':
+    fastllm_lib = ctypes.cdll.LoadLibrary(os.path.join(os.path.split(os.path.realpath(__file__))[0], "libfastllm_tools.dylib"))
 else:
     fastllm_lib = ctypes.cdll.LoadLibrary(os.path.join(os.path.split(os.path.realpath(__file__))[0], "libfastllm_tools.so"))
 

diff --git a/tools/scripts/setup.py b/tools/scripts/setup.py
@@ -10,6 +10,6 @@
     packages = ['fastllm_pytools'],
 
     package_data = {
-        '': ['*.dll', '*.so']
+        '': ['*.dll', '*.so', '*.dylib']
     }
 )