Views
No views yet
| Model | Checkpoint | HumanEval (+) | MBPP (+) |
|---|---|---|---|
| Code Millenials 34B | HF Link | 80.48 (75) | 74.68 (62.9) |
| Code Millenials 13B | HF Link | 76.21 (69.5) | 70.17 (57.6) |
| Code Millenials 3B | HF Link | 56.09 (52.43) | 55.13 (47.11) |
| Code Millenials 1B | HF Link | 51.82 (48.17) | 53.13 (44.61) |
1import torch
2from transformers import AutoTokenizer, AutoModelForCausalLM
3
4tokenizer = AutoTokenizer.from_pretrained("budecosystem/code-millenials-13b")
5model = AutoModelForCausalLM.from_pretrained("budecosystem/code-millenials-13b")
6
7template = """A chat between a curious user and an artificial intelligence assistant. The assistant gives helpful, detailed, and polite answers to the user's questions.
8### Instruction: {instruction} ### Response:"""
9
10instruction = <Your code instruction here>
11
12prompt = template.format(instruction=instruction)
13
14inputs = tokenizer(prompt, return_tensors="pt")
15sample = model.generate(**inputs, max_length=128)
16print(tokenizer.decode(sample[0]))
17| Hyperparameters | Value |
|---|---|
| per_device_train_batch_size | 2 |
| gradient_accumulation_steps | 1 |
| epoch | 3 |
| steps | 34503 |
| learning_rate | 2e-5 |
| lr schedular type | cosine |
| warmup ratio | 0.1 |
| optimizer | adamw |
| fp16 | True |
| GPU | 8 A100 80GB |