library_name: mlx
pipeline_tag: text-generation
tags:

• mlx

mlx-community/DeepSeek-R1-0528-4bit

此模型 mlx-community/DeepSeek-R1-0528-4bit 是从 deepseek-ai/DeepSeek-R1-0528 转换而来，采用了 0.24.1 版本的 mlx-lm 工具。

在 MLX 中使用

pip install mlx-lm  

from mlx_lm import load, generate  

model, tokenizer = load("mlx-community/DeepSeek-R1-0528-4bit")  

prompt = "你好"  

if tokenizer.chat_template is not None:  
    messages = [{"role": "user", "content": prompt}]  
    prompt = tokenizer.apply_chat_template(  
        messages, add_generation_prompt=True  
    )  

response = generate(model, tokenizer, prompt=prompt, verbose=True)