Instructions to use abvijaykumar/distilgpt2-prefix-table-to-text with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- PEFT
How to use abvijaykumar/distilgpt2-prefix-table-to-text with PEFT:
from peft import PeftModel from transformers import AutoModelForCausalLM base_model = AutoModelForCausalLM.from_pretrained("distilgpt2") model = PeftModel.from_pretrained(base_model, "abvijaykumar/distilgpt2-prefix-table-to-text") - Notebooks
- Google Colab
- Kaggle
Commit ·
0acb147
1
Parent(s): e1b14e4
Upload model
Browse files- README.md +9 -0
- adapter_config.json +15 -0
- adapter_model.bin +3 -0
README.md
ADDED
|
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
library_name: peft
|
| 3 |
+
---
|
| 4 |
+
## Training procedure
|
| 5 |
+
|
| 6 |
+
### Framework versions
|
| 7 |
+
|
| 8 |
+
|
| 9 |
+
- PEFT 0.6.0.dev0
|
adapter_config.json
ADDED
|
@@ -0,0 +1,15 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"auto_mapping": null,
|
| 3 |
+
"base_model_name_or_path": "distilgpt2",
|
| 4 |
+
"encoder_hidden_size": 768,
|
| 5 |
+
"inference_mode": true,
|
| 6 |
+
"num_attention_heads": 12,
|
| 7 |
+
"num_layers": 6,
|
| 8 |
+
"num_transformer_submodules": 1,
|
| 9 |
+
"num_virtual_tokens": 20,
|
| 10 |
+
"peft_type": "PREFIX_TUNING",
|
| 11 |
+
"prefix_projection": false,
|
| 12 |
+
"revision": null,
|
| 13 |
+
"task_type": "CAUSAL_LM",
|
| 14 |
+
"token_dim": 768
|
| 15 |
+
}
|
adapter_model.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:babc34713bbb49befff44c525a1cccbd4f21339c6703a54bc3f6569024a4ba07
|
| 3 |
+
size 738109
|