mirror of
https://github.com/hwchase17/langchain
synced 2024-11-10 01:10:59 +00:00
1f9f4d8742
**Description:** This PR adds support for MLX models both chat (i.e., instruct) and llm (i.e., pretrained) types/ **Dependencies:** mlx, mlx_lm, transformers **Twitter handle:** @Prince_Canuma --------- Co-authored-by: Bagatur <baskaryan@gmail.com> Co-authored-by: Bagatur <22008038+baskaryan@users.noreply.github.com>
12 lines
310 B
Python
12 lines
310 B
Python
"""Test MLX Chat wrapper."""
|
|
from importlib import import_module
|
|
|
|
|
|
def test_import_class() -> None:
|
|
"""Test that the class can be imported."""
|
|
module_name = "langchain_community.chat_models.mlx"
|
|
class_name = "ChatMLX"
|
|
|
|
module = import_module(module_name)
|
|
assert hasattr(module, class_name)
|