| 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.0 | |
| 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.0 | |
| name: pass@1 | |
| # mlx-community/granite-20b-code-base-4bit | |
| The Model [mlx-community/granite-20b-code-base-4bit](https://huggingface.co/mlx-community/granite-20b-code-base-4bit) was converted to MLX format from [ibm-granite/granite-20b-code-base](https://huggingface.co/ibm-granite/granite-20b-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-20b-code-base-4bit") | |
| response = generate(model, tokenizer, prompt="hello", verbose=True) | |
| ``` | |