Yi-1.5-6B-Chat-4bit / README.md
prince-canuma's picture
Update README.md
0177c9a verified
metadata
license: apache-2.0
tags:
  - mlx

mlx-community/Yi-1.5-6B-Chat-4bit

The Model mlx-community/Yi-1.5-6B-Chat-4bit was converted to MLX format from 01-ai/Yi-1.5-6B-Chat using mlx-lm version 0.13.0.

Model added by Prince Canuma.

Use with mlx

pip install mlx-lm
from mlx_lm import load, generate

model, tokenizer = load("mlx-community/Yi-1.5-6B-Chat-4bit")
response = generate(model, tokenizer, prompt="hello", verbose=True)