Instructions to use anirvankrishna/TaskB_Prefix_Finetune with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use anirvankrishna/TaskB_Prefix_Finetune with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("anirvankrishna/TaskB_Prefix_Finetune", device_map="auto") - Notebooks
- Google Colab
- Kaggle
File size: 442 Bytes
63a46a4 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 | {
"auto_mapping": null,
"base_model_name_or_path": "openai-community/gpt2-medium",
"encoder_hidden_size": 1024,
"inference_mode": true,
"modules_to_save": null,
"num_attention_heads": 16,
"num_layers": 24,
"num_transformer_submodules": 1,
"num_virtual_tokens": 10,
"peft_type": "PREFIX_TUNING",
"peft_version": "0.18.1",
"prefix_projection": false,
"revision": null,
"task_type": "CAUSAL_LM",
"token_dim": 1024
} |