Instructions to use andstor/Salesforce-codegen-350M-multi-unit-test-prompt-tuning with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- PEFT
How to use andstor/Salesforce-codegen-350M-multi-unit-test-prompt-tuning with PEFT:
from peft import PeftModel from transformers import AutoModelForCausalLM base_model = AutoModelForCausalLM.from_pretrained("Salesforce/codegen-350M-multi") model = PeftModel.from_pretrained(base_model, "andstor/Salesforce-codegen-350M-multi-unit-test-prompt-tuning") - Notebooks
- Google Colab
- Kaggle
File size: 248 Bytes
7d2892f | 1 2 3 4 5 6 7 8 9 | {
"epoch": 2.9991668980838657,
"total_flos": 2.317351723938611e+16,
"train_loss": 0.7935600814112911,
"train_runtime": 1657.45,
"train_samples": 7202,
"train_samples_per_second": 13.036,
"train_steps_per_second": 0.815
} |