Instructions to use belweave/kai-2 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

Libraries

How to use belweave/kai-2 with Transformers:

# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("text-generation", model="belweave/kai-2")
messages = [
    {"role": "user", "content": "Who are you?"},
]
pipe(messages)

# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("belweave/kai-2")
model = AutoModelForCausalLM.from_pretrained("belweave/kai-2")
messages = [
    {"role": "user", "content": "Who are you?"},
]
inputs = tokenizer.apply_chat_template(
	messages,
	add_generation_prompt=True,
	tokenize=True,
	return_dict=True,
	return_tensors="pt",
).to(model.device)

outputs = model.generate(**inputs, max_new_tokens=40)
print(tokenizer.decode(outputs[0][inputs["input_ids"].shape[-1]:]))

MLX

How to use belweave/kai-2 with MLX:

# Make sure mlx-lm is installed
# pip install --upgrade mlx-lm

# Generate text with mlx-lm
from mlx_lm import load, generate

model, tokenizer = load("belweave/kai-2")

prompt = "Write a story about Einstein"
messages = [{"role": "user", "content": prompt}]
prompt = tokenizer.apply_chat_template(
    messages, add_generation_prompt=True
)

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

llama-cpp-python

How to use belweave/kai-2 with llama-cpp-python:

# !pip install llama-cpp-python

from llama_cpp import Llama

llm = Llama.from_pretrained(
	repo_id="belweave/kai-2",
	filename="kai-2-Q4_K_M.gguf",
)

llm.create_chat_completion(
	messages = [
		{
			"role": "user",
			"content": "What is the capital of France?"
		}
	]
)

Notebooks
Google Colab
Kaggle
Local Apps Settings

llama.cpp

How to use belweave/kai-2 with llama.cpp:

Install from brew

brew install llama.cpp
# Start a local OpenAI-compatible server with a web UI:
llama-server -hf belweave/kai-2:Q4_K_M
# Run inference directly in the terminal:
llama-cli -hf belweave/kai-2:Q4_K_M

Install from WinGet (Windows)

winget install llama.cpp
# Start a local OpenAI-compatible server with a web UI:
llama-server -hf belweave/kai-2:Q4_K_M
# Run inference directly in the terminal:
llama-cli -hf belweave/kai-2:Q4_K_M

Use pre-built binary

# Download pre-built binary from:
# https://github.com/ggerganov/llama.cpp/releases
# Start a local OpenAI-compatible server with a web UI:
./llama-server -hf belweave/kai-2:Q4_K_M
# Run inference directly in the terminal:
./llama-cli -hf belweave/kai-2:Q4_K_M

Build from source code

git clone https://github.com/ggerganov/llama.cpp.git
cd llama.cpp
cmake -B build
cmake --build build -j --target llama-server llama-cli
# Start a local OpenAI-compatible server with a web UI:
./build/bin/llama-server -hf belweave/kai-2:Q4_K_M
# Run inference directly in the terminal:
./build/bin/llama-cli -hf belweave/kai-2:Q4_K_M

Use Docker

docker model run hf.co/belweave/kai-2:Q4_K_M

LM Studio
Jan

vLLM

How to use belweave/kai-2 with vLLM:

Install from pip and serve model

# Install vLLM from pip:
pip install vllm
# Start the vLLM server:
vllm serve "belweave/kai-2"
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:8000/v1/chat/completions" \
	-H "Content-Type: application/json" \
	--data '{
		"model": "belweave/kai-2",
		"messages": [
			{
				"role": "user",
				"content": "What is the capital of France?"
			}
		]
	}'

Use Docker

docker model run hf.co/belweave/kai-2:Q4_K_M

SGLang

How to use belweave/kai-2 with SGLang:

Install from pip and serve model

# Install SGLang from pip:
pip install sglang
# Start the SGLang server:
python3 -m sglang.launch_server \
    --model-path "belweave/kai-2" \
    --host 0.0.0.0 \
    --port 30000
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:30000/v1/chat/completions" \
	-H "Content-Type: application/json" \
	--data '{
		"model": "belweave/kai-2",
		"messages": [
			{
				"role": "user",
				"content": "What is the capital of France?"
			}
		]
	}'

Use Docker images

docker run --gpus all \
    --shm-size 32g \
    -p 30000:30000 \
    -v ~/.cache/huggingface:/root/.cache/huggingface \
    --env "HF_TOKEN=<secret>" \
    --ipc=host \
    lmsysorg/sglang:latest \
    python3 -m sglang.launch_server \
        --model-path "belweave/kai-2" \
        --host 0.0.0.0 \
        --port 30000
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:30000/v1/chat/completions" \
	-H "Content-Type: application/json" \
	--data '{
		"model": "belweave/kai-2",
		"messages": [
			{
				"role": "user",
				"content": "What is the capital of France?"
			}
		]
	}'

Ollama
How to use belweave/kai-2 with Ollama:
```
ollama run hf.co/belweave/kai-2:Q4_K_M
```

Unsloth Studio

How to use belweave/kai-2 with Unsloth Studio:

Install Unsloth Studio (macOS, Linux, WSL)

curl -fsSL https://unsloth.ai/install.sh | sh
# Run unsloth studio
unsloth studio -H 0.0.0.0 -p 8888
# Then open http://localhost:8888 in your browser
# Search for belweave/kai-2 to start chatting

Install Unsloth Studio (Windows)

irm https://unsloth.ai/install.ps1 | iex
# Run unsloth studio
unsloth studio -H 0.0.0.0 -p 8888
# Then open http://localhost:8888 in your browser
# Search for belweave/kai-2 to start chatting

Using HuggingFace Spaces for Unsloth

# No setup required
# Open https://huggingface.co/spaces/unsloth/studio in your browser
# Search for belweave/kai-2 to start chatting

How to use belweave/kai-2 with Pi:

Start the MLX server

# Install MLX LM:
uv tool install mlx-lm
# Start a local OpenAI-compatible server:
mlx_lm.server --model "belweave/kai-2"

Configure the model in Pi

# Install Pi:
npm install -g @mariozechner/pi-coding-agent
# Add to ~/.pi/agent/models.json:
{
  "providers": {
    "mlx-lm": {
      "baseUrl": "http://localhost:8080/v1",
      "api": "openai-completions",
      "apiKey": "none",
      "models": [
        {
          "id": "belweave/kai-2"
        }
      ]
    }
  }
}

Run Pi

# Start Pi in your project directory:
pi

Hermes Agent new

How to use belweave/kai-2 with Hermes Agent:

Start the MLX server

# Install MLX LM:
uv tool install mlx-lm
# Start a local OpenAI-compatible server:
mlx_lm.server --model "belweave/kai-2"

Configure Hermes

# Install Hermes:
curl -fsSL https://hermes-agent.nousresearch.com/install.sh | bash
hermes setup
# Point Hermes at the local server:
hermes config set model.provider custom
hermes config set model.base_url http://127.0.0.1:8080/v1
hermes config set model.default belweave/kai-2

Run Hermes

hermes

MLX LM

How to use belweave/kai-2 with MLX LM:

Generate or start a chat session

# Install MLX LM
uv tool install mlx-lm
# Interactive chat REPL
mlx_lm.chat --model "belweave/kai-2"

Run an OpenAI-compatible server

# Install MLX LM
uv tool install mlx-lm
# Start the server
mlx_lm.server --model "belweave/kai-2"
# Calling the OpenAI-compatible server with curl
curl -X POST "http://localhost:8000/v1/chat/completions" \
   -H "Content-Type: application/json" \
   --data '{
     "model": "belweave/kai-2",
     "messages": [
       {"role": "user", "content": "Hello"}
     ]
   }'

Docker Model Runner
How to use belweave/kai-2 with Docker Model Runner:
```
docker model run hf.co/belweave/kai-2:Q4_K_M
```

Lemonade

How to use belweave/kai-2 with Lemonade:

Pull the model

# Download Lemonade from https://lemonade-server.ai/
lemonade pull belweave/kai-2:Q4_K_M

Run and chat with the model

lemonade run user.kai-2-Q4_K_M

List all available models

lemonade list

preethamBelweave commited on 30 days ago

Commit

cb80baf

verified ·

1 Parent(s): 9876a12

Upload README.md with huggingface_hub

Browse files

Files changed (1) hide show

README.md +29 -0

README.md CHANGED Viewed

@@ -182,6 +182,35 @@ response = generate(
 print(response)
 ```
 ## Model Architecture Details
 - **Hidden Size:** 3,584

 print(response)
 ```
+## GGUF Quantizations
+For local inference with **LM Studio**, **llama.cpp**, or **Ollama**, download a GGUF variant:
+| Quantization | Size | BPW | Best For |
+|--------------|------|-----|----------|
+| [Q4_K_M](https://huggingface.co/belweave/kai-2/blob/main/kai-2-Q4_K_M.gguf) | ~4.7 GB | 4.91 | Speed/quality balance (recommended) |
+| [Q5_K_M](https://huggingface.co/belweave/kai-2/blob/main/kai-2-Q5_K_M.gguf) | ~5.4 GB | 5.71 | Slightly higher quality |
+| [Q8_0](https://huggingface.co/belweave/kai-2/blob/main/kai-2-Q8_0.gguf) | ~8.1 GB | 8.50 | Near-lossless quality |
+### LM Studio
+1. Download any `.gguf` file above
+2. Open LM Studio → **My Models** → **Load from Disk**
+3. Select the downloaded `.gguf` file
+4. In **Chat Settings**, ensure the system prompt is left **empty** (the model's chat template already handles identity)
+### llama.cpp / Ollama
+```bash
+# llama.cpp
+./llama-cli -m kai-2-Q4_K_M.gguf -p "Who are you?"
+# Ollama (create a Modelfile)
+echo 'FROM ./kai-2-Q4_K_M.gguf' > Modelfile
+ollama create kai-2 -f Modelfile
+ollama run kai-2
+```
 ## Model Architecture Details
 - **Hidden Size:** 3,584