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-34b-code-base | |
results: | |
- task: | |
type: text-generation | |
dataset: | |
name: MBPP | |
type: mbpp | |
metrics: | |
- type: pass@1 | |
value: 47.2 | |
name: pass@1 | |
- task: | |
type: text-generation | |
dataset: | |
name: MBPP+ | |
type: evalplus/mbppplus | |
metrics: | |
- type: pass@1 | |
value: 53.1 | |
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: 54.9 | |
name: pass@1 | |
- type: pass@1 | |
value: 61.6 | |
name: pass@1 | |
- type: pass@1 | |
value: 40.2 | |
name: pass@1 | |
- type: pass@1 | |
value: 50.0 | |
name: pass@1 | |
- type: pass@1 | |
value: 39.6 | |
name: pass@1 | |
- type: pass@1 | |
value: 42.7 | |
name: pass@1 | |
- type: pass@1 | |
value: 26.2 | |
name: pass@1 | |
- type: pass@1 | |
value: 47.0 | |
name: pass@1 | |
- type: pass@1 | |
value: 26.8 | |
name: pass@1 | |
- type: pass@1 | |
value: 36.6 | |
name: pass@1 | |
- type: pass@1 | |
value: 25.0 | |
name: pass@1 | |
- type: pass@1 | |
value: 20.1 | |
name: pass@1 | |
- type: pass@1 | |
value: 30.5 | |
name: pass@1 | |
- type: pass@1 | |
value: 40.9 | |
name: pass@1 | |
- type: pass@1 | |
value: 34.1 | |
name: pass@1 | |
- type: pass@1 | |
value: 39.0 | |
name: pass@1 | |
- type: pass@1 | |
value: 12.2 | |
name: pass@1 | |
# mlx-community/granite-34b-code-base-8bit | |
The Model [mlx-community/granite-34b-code-base-8bit](https://huggingface.co/mlx-community/granite-34b-code-base-8bit) was converted to MLX format from [ibm-granite/granite-34b-code-base](https://huggingface.co/ibm-granite/granite-34b-code-base) using mlx-lm version **0.13.0**. | |
## Use with mlx | |
```bash | |
pip install mlx-lm | |
``` | |
```python | |
from mlx_lm import load, generate | |
model, tokenizer = load("mlx-community/granite-34b-code-base-8bit") | |
response = generate(model, tokenizer, prompt="hello", verbose=True) | |
``` | |