mirror of
https://github.com/ml-explore/mlx-examples.git
synced 2025-06-25 01:41:19 +08:00
954aa50c54
* LoRA: Improve validation error for LoRA layer count exceeding model layer This commit enhances the error handling when the specified LoRA layer count exceeds the total number of layers in the model. It clarifies the error message to provide actionable feedback for users, guiding them to adjust their input parameters accordingly. * format + nits --------- Co-authored-by: Awni Hannun <awni@apple.com>
24 lines
763 B
Python
24 lines
763 B
Python
import sys
|
|
from pathlib import Path
|
|
|
|
import pkg_resources
|
|
from setuptools import setup
|
|
|
|
with open(Path(__file__).parent / "mlx_lm/requirements.txt") as fid:
|
|
requirements = [str(r) for r in pkg_resources.parse_requirements(fid)]
|
|
setup(
|
|
name="mlx-lm",
|
|
version="0.0.10",
|
|
description="LLMs on Apple silicon with MLX and the Hugging Face Hub",
|
|
long_description=open("README.md", encoding="utf-8").read(),
|
|
long_description_content_type="text/markdown",
|
|
readme="README.md",
|
|
author_email="mlx@group.apple.com",
|
|
author="MLX Contributors",
|
|
url="https://github.com/ml-explore/mlx-examples",
|
|
license="MIT",
|
|
install_requires=requirements,
|
|
packages=["mlx_lm", "mlx_lm.models", "mlx_lm.tuner"],
|
|
python_requires=">=3.8",
|
|
)
|