How to use from the
Use from the
llama-cpp-python library
# !pip install llama-cpp-python

from llama_cpp import Llama

llm = Llama.from_pretrained(
	repo_id="Tekimax/granite-ml-coder-GGUF",
	filename="granite-ml-coder-Q4_K_M.gguf",
)
llm.create_chat_completion(
	messages = [
		{
			"role": "user",
			"content": "What is the capital of France?"
		}
	]
)

granite-ml-coder-GGUF

GGUF builds of Tekimax/granite-ml-coder — a compact Python / machine-learning coding assistant fine-tuned from ibm-granite/granite-3.1-1b-a400m-instruct. These run on CPU and Apple Silicon (Metal) via llama.cpp, Ollama, and LM Studio.

Files

File Bits Size Notes
granite-ml-coder-Q4_K_M.gguf 4-bit ~378 MB recommended — loads fast, near-full quality

Run it

Ollama (from the public registry)

ollama run tekimaxllc/granite-ml-coder "Write a sklearn pipeline for the iris dataset"

Ollama (from this GGUF directly)

ollama run hf.co/Tekimax/granite-ml-coder-GGUF

llama.cpp

llama-cli -m granite-ml-coder-Q4_K_M.gguf \
  -p "Write a Keras autoencoder for network-traffic anomaly detection"

LM Studio

Search for Tekimax/granite-ml-coder-GGUF, download the Q4_K_M file, and chat.

Suggested system prompt

You are an expert Python machine-learning engineer. Write correct, runnable code,
explain each pipeline step, watch for overfitting and how gradient descent
converges, and recommend the best model or formula for the task.

Intended use & limitations

Good for drafting Python ML code (scikit-learn, pandas, NumPy, Keras) and explaining ML concepts, fully offline. It's a 1B model — treat output as a fast first draft and verify before use. See the full model card for training details and limitations.

License

Apache-2.0 (inherited from ibm-granite/granite-3.1-1b-a400m-instruct).

Downloads last month
177
GGUF
Model size
1B params
Architecture
granitemoe
Hardware compatibility
Log In to add your hardware

4-bit

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

Model tree for Tekimax/granite-ml-coder-GGUF

Space using Tekimax/granite-ml-coder-GGUF 1