How to use from the
Use from the
MLX library
# Make sure mlx-lm is installed
# pip install --upgrade mlx-lm
# if on a CUDA device, also pip install mlx[cuda]

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

model, tokenizer = load("ipetrukha/codegemma-1.1-2b-4bit")

prompt = "Once upon a time in"
text = generate(model, tokenizer, prompt=prompt, verbose=True)

ipetrukha/codegemma-1.1-2b-4bit

The Model ipetrukha/codegemma-1.1-2b-4bit was converted to MLX format from google/codegemma-1.1-2b using mlx-lm version 0.16.1.

Use with mlx

pip install mlx-lm
from mlx_lm import load, generate

model, tokenizer = load("ipetrukha/codegemma-1.1-2b-4bit")
response = generate(model, tokenizer, prompt="hello", verbose=True)
Downloads last month
6
Safetensors
Model size
0.4B params
Tensor type
F16
·
U32
·
MLX
Hardware compatibility
Log In to add your hardware

Quantized

Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support