Instructions to use fals3/bigcode-starcoderbase-unit-test-prompt-tuning with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- PEFT
How to use fals3/bigcode-starcoderbase-unit-test-prompt-tuning with PEFT:
from peft import PeftModel from transformers import AutoModelForCausalLM base_model = AutoModelForCausalLM.from_pretrained("bigcode/starcoderbase") model = PeftModel.from_pretrained(base_model, "fals3/bigcode-starcoderbase-unit-test-prompt-tuning") - Notebooks
- Google Colab
- Kaggle
File size: 234 Bytes
0bfb53d | 1 2 3 4 5 6 7 8 9 | {
"epoch": 3.0,
"total_flos": 1.0021240006560973e+18,
"train_loss": 1.5448391757168614,
"train_runtime": 5139.738,
"train_samples": 7279,
"train_samples_per_second": 4.249,
"train_steps_per_second": 0.266
} |