# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("ipetrukha/codegemma-1.1-2b-4bit")
model = AutoModelForCausalLM.from_pretrained("ipetrukha/codegemma-1.1-2b-4bit")Quick Links
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
- 9
Model size
0.4B params
Tensor type
F16
·
U32 ·
Hardware compatibility
Log In to add your hardware
Quantized
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-generation", model="ipetrukha/codegemma-1.1-2b-4bit")