Skip to content

Commit

Permalink
Add mlc_serve as a package
Browse files Browse the repository at this point in the history
  • Loading branch information
mehrdadh committed Jan 19, 2024
1 parent b5f3742 commit 6e8cf90
Showing 1 changed file with 6 additions and 2 deletions.
8 changes: 6 additions & 2 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,6 +22,10 @@ def git_describe_version(original_version):

__version__ = git_describe_version(None)

def find_mlc_serve():
packages = find_packages(where="serve")
return [package for package in packages if package.startswith("mlc_serve")]

setup(
name="mlc_llm",
version=__version__,
Expand All @@ -39,8 +43,8 @@ def git_describe_version(original_version):
],
keywords="machine learning",
zip_safe=False,
packages=find_packages(),
package_dir={"mlc_llm": "mlc_llm"},
packages=find_packages() + find_mlc_serve(),
package_dir={"mlc_llm": "mlc_llm", "mlc_serve": "serve/mlc_serve"},
install_requires=["numpy", "torch", "transformers", "scipy", "timm"],
entry_points={"console_scripts": ["mlc_llm_build = mlc_llm.build:main"]},
distclass=Distribution,
Expand Down

0 comments on commit 6e8cf90

Please sign in to comment.