metadata
license: apache-2.0
library_name: transformers
tags:
- code
- granite
- mlx
datasets:
- codeparrot/github-code-clean
- bigcode/starcoderdata
- open-web-math/open-web-math
- math-ai/StackMathQA
metrics:
- code_eval
pipeline_tag: text-generation
inference: true
model-index:
- name: granite-20b-code-base
results:
- task:
type: text-generation
dataset:
name: MBPP
type: mbpp
metrics:
- type: pass@1
value: 43.8
name: pass@1
- task:
type: text-generation
dataset:
name: MBPP+
type: evalplus/mbppplus
metrics:
- type: pass@1
value: 51.6
name: pass@1
- task:
type: text-generation
dataset:
name: HumanEvalSynthesis(Python)
type: bigcode/humanevalpack
metrics:
- type: pass@1
value: 48.2
name: pass@1
- type: pass@1
value: 50
name: pass@1
- type: pass@1
value: 59.1
name: pass@1
- type: pass@1
value: 32.3
name: pass@1
- type: pass@1
value: 40.9
name: pass@1
- type: pass@1
value: 35.4
name: pass@1
- type: pass@1
value: 17.1
name: pass@1
- type: pass@1
value: 18.3
name: pass@1
- type: pass@1
value: 23.2
name: pass@1
- type: pass@1
value: 10.4
name: pass@1
- type: pass@1
value: 25.6
name: pass@1
- type: pass@1
value: 18.3
name: pass@1
- type: pass@1
value: 23.2
name: pass@1
- type: pass@1
value: 23.8
name: pass@1
- type: pass@1
value: 14.6
name: pass@1
- type: pass@1
value: 26.2
name: pass@1
- type: pass@1
value: 15.2
name: pass@1
- type: pass@1
value: 3
name: pass@1
mlx-community/granite-20b-code-base-4bit
The Model mlx-community/granite-20b-code-base-4bit was converted to MLX format from ibm-granite/granite-20b-code-base using mlx-lm version 0.13.0.
Use with mlx
pip install mlx-lm
from mlx_lm import load, generate
model, tokenizer = load("mlx-community/granite-20b-code-base-4bit")
response = generate(model, tokenizer, prompt="hello", verbose=True)