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