Total runs:
123.0K
Run Growth:
90.5K
Growth Rate:
73.59%
Sponsored by
APIMart
bigcode / astraios-3b-ia3
Model's Last Updated: January 02 2024
Introduction of astraios-3b-ia3
Model Details of astraios-3b-ia3
Astraios: Parameter-Efficient Instruction Tuning Code Large Language Models
Table of Contents
Model Summary
Astraios-3B-IA3 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-3b-ia3"
checkpoint = "bigcode/starcoderbase-3b"
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-3b-ia3 on huggingface.co
9
Total runs
0
24-hour runs
-1
3-day runs
1
7-day runs
4
30-day runs
More Information About astraios-3b-ia3 huggingface.co Model
Url of astraios-3b-ia3
astraios-3b-ia3 huggingface.co Url
Provider of astraios-3b-ia3 huggingface.co
Other API from bigcode
Updated:June 01 2023
Total runs:
86.7K
Run Growth:
-18.0K
Growth Rate:
-20.82%
Total runs:
41.9K
Run Growth:
-26.8K
Growth Rate:
-63.93%
Total runs:
4.7K
Run Growth:
499
Growth Rate:
10.52%
Total runs:
896
Run Growth:
-1.3K
Growth Rate:
-145.87%
Total runs:
343
Run Growth:
-582
Growth Rate:
-169.68%
Total runs:
208
Run Growth:
201
Growth Rate:
96.63%
Total runs:
199
Run Growth:
191
Growth Rate:
95.98%
Total runs:
178
Run Growth:
172
Growth Rate:
96.63%
Total runs:
109
Run Growth:
-419
Growth Rate:
-384.40%
Total runs:
67
Run Growth:
52
Growth Rate:
77.61%
Total runs:
22
Run Growth:
18
Growth Rate:
81.82%
Total runs:
9
Run Growth:
5
Growth Rate:
55.56%
Total runs:
8
Run Growth:
-2
Growth Rate:
-25.00%
Total runs:
7
Run Growth:
-4
Growth Rate:
-57.14%
Total runs:
6
Run Growth:
-4
Growth Rate:
-66.67%
Total runs:
5
Run Growth:
-5
Growth Rate:
-100.00%
Total runs:
5
Run Growth:
-3
Growth Rate:
-60.00%
Total runs:
4
Run Growth:
-1
Growth Rate:
-25.00%
Total runs:
3
Run Growth:
-9
Growth Rate:
-300.00%
Total runs:
3
Run Growth:
-3
Growth Rate:
-100.00%
Total runs:
3
Run Growth:
-5
Growth Rate:
-166.67%
Total runs:
3
Run Growth:
-7
Growth Rate:
-233.33%
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%