许可证: apache-2.0
语言:

• 英文
  数据集:
• allenai/tulu-3-sft-olmo-2-mixture
  基础模型: allenai/OLMo-2-0325-32B-Instruct
  任务标签: 文本生成
  库名称: transformers
  标签:
• mlx

mlx-community/OLMo-2-0325-32B-Instruct-4bit

模型 mlx-community/OLMo-2-0325-32B-Instruct-4bit 是从 allenai/OLMo-2-0325-32B-Instruct 使用 mlx-lm 0.22.0 版本转换至 MLX 格式的。

在 MLX 中使用

pip install mlx-lm  

from mlx_lm import load, generate  

model, tokenizer = load("mlx-community/OLMo-2-0325-32B-Instruct-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)