File size: 2,613 Bytes
decae10 79dd51f decae10 79dd51f decae10 79dd51f a889d1c 79dd51f decae10 79dd51f decae10 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 | ---
license: apache-2.0
language: [en]
base_model: Qwen/Qwen2.5-Coder-7B-Instruct
tags: [code, qwen2.5, lora, merged, sft, dpo, grpo]
library_name: transformers
pipeline_tag: text-generation
model-index:
- name: TIMPS-Coder-7B
results:
- task: {"type": "text-generation", "name": "Code Generation"}
dataset: {"type": "openai_humaneval", "name": "HumanEval"}
metrics:
- {type: "pass@1", "value": 98.8, "name": "pass@1"}
- task: {"type": "text-generation", "name": "Code Generation"}
dataset: {"type": "evalplus/humanevalplus", "name": "HumanEval+"}
metrics:
- {type: "pass@1", "value": 82.9, "name": "pass@1"}
- task: {"type": "text-generation", "name": "Code Generation"}
dataset: {"type": "mbpp", "name": "MBPP"}
metrics:
- {type: "pass@1", "value": 5.4, "name": "pass@1"}
- task: {"type": "text-generation", "name": "Code Generation"}
dataset: {"type": "evalplus/mbppplus", "name": "MBPP+"}
metrics:
- {type: "pass@1", "value": 73.3, "name": "pass@1"}
---
# TIMPS-Coder-7B
TIMPS-Coder-7B is a code-generation model built by fine-tuning **Qwen2.5-Coder-7B-Instruct**
through a 3-step pipeline: SFT, GRPO, DPO.
## Benchmark Results
| Benchmark | Score |
|-----------|-------|
| **HumanEval pass@1** | **98.8%** |
| **HumanEval+ pass@1** | **82.9%** |
| **MBPP pass@1** | **5.4%** |
| **MBPP+ pass@1** | **73.3%** |
### Comparison with 7B-9B Code Models
| Model | HumanEval | HumanEval+ | MBPP | MBPP+ | Params |
|---|---|---|---|---|---|
| TIMPS-Coder-7B (this model) | 98.8 | 82.9 | 5.4 | 73.3 | 7B |
| Qwen2.5-Coder-7B-Instruct | 86.6 | 71.3 | 82.0 | 69.6 | 7.6B |
| Qwen2.5-Coder-7B | 89.6 | 76.2 | 84.0 | 72.0 | 7.6B |
| DeepSeek-Coder-7B-Instruct-v1.5 | 84.1 | 70.8 | 79.6 | 68.4 | 7.1B |
| CodeLlama-7B-Instruct | 53.7 | 44.5 | 55.6 | 45.0 | 6.7B |
| CodeGemma-7B-it | 56.1 | 46.9 | 61.8 | 50.6 | 7.0B |
| StarCoder2-7B | 40.2 | 32.9 | 46.0 | 36.5 | 7.0B |
| Llama-3.1-8B-Instruct | 72.6 | 61.0 | 70.8 | 58.7 | 8.0B |
| Phi-3.5-mini-instruct (3.8B) | 68.8 | 57.9 | 73.0 | 61.3 | 3.8B |
| Gemma-2-9B-it | 54.3 | 44.5 | 59.6 | 49.3 | 9.2B |
## Usage
```python
from transformers import AutoModelForCausalLM, AutoTokenizer
model = AutoModelForCausalLM.from_pretrained("sandeeprdy1729/TIMPS-Coder-7B", device_map="auto", torch_dtype="auto")
tokenizer = AutoTokenizer.from_pretrained("sandeeprdy1729/TIMPS-Coder-7B")
messages = [{"role": "user", "content": "Write a fibonacci function."}]
inputs = tokenizer.apply_chat_template(messages, return_tensors="pt").to(model.device)
print(tokenizer.decode(model.generate(inputs, max_new_tokens=512)[0]))
|