ailexleon's picture
Update README.md
6bdf54c verified
|
Raw
History Blame Contribute Delete
864 Bytes
metadata
base_model: TheDrummer/Behemoth-128B-v3
language: en
library_name: mlx
pipeline_tag: text-generation
tags:
  - mlx
  - rp
  - roleplay
  - creative
  - writer
  - creative writing
  - conversational
  - character-rp
  - storytelling

ailexleon/Behemoth-128B-v3-mlx-5Bit

Converted to MLX format from TheDrummer/Behemoth-128B-v3 using mlx-lm version 0.31.3.

Use with mlx

pip install mlx-lm
from mlx_lm import load, generate

model, tokenizer = load("ailexleon/Behemoth-128B-v3-mlx-5Bit")

prompt = "hello"

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

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