Instructions to use andstor/bigcode-starcoder2-3b-unit-test-prompt-tuning with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- PEFT
How to use andstor/bigcode-starcoder2-3b-unit-test-prompt-tuning with PEFT:
from peft import PeftModel from transformers import AutoModelForCausalLM base_model = AutoModelForCausalLM.from_pretrained("bigcode/starcoder2-3b") model = PeftModel.from_pretrained(base_model, "andstor/bigcode-starcoder2-3b-unit-test-prompt-tuning") - Notebooks
- Google Colab
- Kaggle
File size: 234 Bytes
7f31a51 | 1 2 3 4 5 6 7 8 9 | {
"epoch": 3.0,
"total_flos": 1.9035319978426368e+17,
"train_loss": 2.436253024195577,
"train_runtime": 2343.5965,
"train_samples": 7280,
"train_samples_per_second": 9.319,
"train_steps_per_second": 0.582
} |