Total runs:
231.1K
Run Growth:
-37.3K
Growth Rate:
-16.13%
Sponsored by
Tripo AI
bigcode / astraios-7b-parallel
Model's Last Updated: January 02 2024
Introduction of astraios-7b-parallel
Model Details of astraios-7b-parallel
Astraios: Parameter-Efficient Instruction Tuning Code Large Language Models
Table of Contents
Model Summary
Astraios-7B-Parallel Adapter is an instruction tuned model with 15.5B parameters created by finetuning StarCoderBase on CommitPackFT & OASST as described in the Astraios paper.
- Repository: bigcode-project/astraios
- Paper: Astraios: Parameter-Efficient Instruction Tuning Code Large Language Models
- Languages: 80+ Programming languages
-
✨Astraios:
Data CommitPackFT+OASST Filtered version of CommitPack and OASST for high-quality commit messages that resemble instructions Model Astraios-1B Collection of StarCoderBase-1B models instruction tuned on CommitPackFT + OASST with different tuning methods Astraios-3B Collection of StarCoderBase-3B (3B parameters) models instruction tuned on CommitPackFT + OASST with different tuning methods Astraios-7B Collection of StarCoderBase-7B (7B parameters) models instruction tuned on CommitPackFT + OASST with different tuning methods Astraios-16B Collection of StarCoderBase-16B (16B parameters) models instruction tuned on CommitPackFT + OASST with different tuning methods Evaluation BigCloneBench Dataset for clone detection; We use 2,000 samples for evaluation Devign Dataset for defect detection; We use 2,000 samples for evaluation HumanEvalPack Extension of OpenAI's HumanEval to cover 3 scenarios across 6 languages ReCode Dataset for the robustness of code generation, covering 4 variants Asleep At The Keyboard Datasets for security of code generation; We use DoW for evaluation
Use
Intended use
The model follows instructions provided in the input. You should always preface your input with "Question: " and finish it with "Answer:", for example: "Question: Please write a function in Python that performs bubble sort.
Answer:"
Feel free to share your generations in the Community tab!
Generation
# pip install -q transformers
# pip install -e git+https://github.com/bigcode-project/astraios#subdirectory=peft
from peft import PeftModel
from transformers import AutoModelForCausalLM, AutoTokenizer
peft_checkpoint = "bigcode/astraios-7b-parallel"
checkpoint = "bigcode/starcoderbase-7b"
model = AutoModelForCausalLM.from_pretrained(checkpoint)
model = PeftModel.from_pretrained(model, peft_checkpoint)
device = "cuda" # for GPU usage or "cpu" for CPU usage
tokenizer = AutoTokenizer.from_pretrained(checkpoint)
model = AutoModelForCausalLM.from_pretrained(checkpoint).to(device)
inputs = tokenizer.encode("Question: Please write a function in Python that performs bubble sort.
Answer:", return_tensors="pt").to(device)
outputs = model.generate(inputs)
print(tokenizer.decode(outputs[0]))
Training
Model
- Architecture: GPT-2 model with multi-query attention and Fill-in-the-Middle objective
- Steps: 250k pretraining & 200 instruction tuning
- Precision: fp32
Hardware
-
Pretraining:
- GPUs: 512 Tesla A100
- Training time: 24 days
-
Instruction tuning:
- GPUs: 8 Tesla A100
Software
- Orchestration: Megatron-LM/Transformers
- Neural networks: PyTorch
Citation
Runs of bigcode astraios-7b-parallel on huggingface.co
24
Total runs
19
24-hour runs
19
3-day runs
16
7-day runs
20
30-day runs
More Information About astraios-7b-parallel huggingface.co Model
Url of astraios-7b-parallel
astraios-7b-parallel huggingface.co Url
Provider of astraios-7b-parallel huggingface.co
Other API from bigcode
Updated:June 01 2023
Total runs:
172.4K
Run Growth:
-12.3K
Growth Rate:
-7.14%
Total runs:
35.9K
Run Growth:
-9.9K
Growth Rate:
-27.56%
Total runs:
1.9K
Run Growth:
-2.0K
Growth Rate:
-103.63%
Total runs:
561
Run Growth:
-553
Growth Rate:
-98.57%
Total runs:
293
Run Growth:
286
Growth Rate:
98.62%
Total runs:
200
Run Growth:
10
Growth Rate:
5.00%
Total runs:
190
Run Growth:
12
Growth Rate:
6.32%
Total runs:
157
Run Growth:
-3
Growth Rate:
-1.91%
Total runs:
142
Run Growth:
-295
Growth Rate:
-207.75%
Total runs:
94
Run Growth:
60
Growth Rate:
63.83%
Total runs:
3
Run Growth:
1
Growth Rate:
33.33%
Total runs:
2
Run Growth:
1
Growth Rate:
50.00%
Total runs:
2
Run Growth:
-4
Growth Rate:
-200.00%
Total runs:
2
Run Growth:
2
Growth Rate:
100.00%
Total runs:
1
Run Growth:
-4
Growth Rate:
-400.00%
Total runs:
1
Run Growth:
-1
Growth Rate:
-100.00%
Total runs:
0
Run Growth:
0
Growth Rate:
0.00%
Total runs:
0
Run Growth:
0
Growth Rate:
0.00%
Total runs:
0
Run Growth:
0
Growth Rate:
0.00%
Total runs:
0
Run Growth:
0
Growth Rate:
0.00%
Total runs:
0
Run Growth:
0
Growth Rate:
0.00%