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
| { | |
| "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 | |
| } |