Instructions to use fals3/bigcode-starcoder2-7b-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-starcoder2-7b-unit-test-prompt-tuning with PEFT:
from peft import PeftModel from transformers import AutoModelForCausalLM base_model = AutoModelForCausalLM.from_pretrained("bigcode/starcoder2-7b") model = PeftModel.from_pretrained(base_model, "fals3/bigcode-starcoder2-7b-unit-test-prompt-tuning") - Notebooks
- Google Colab
- Kaggle
| { | |
| "epoch": 3.0, | |
| "total_flos": 4.592889669447844e+17, | |
| "train_loss": 1.2525357378708137, | |
| "train_runtime": 3291.1671, | |
| "train_samples": 7280, | |
| "train_samples_per_second": 6.636, | |
| "train_steps_per_second": 0.415 | |
| } |