Instructions to use bigcode/astraios-7b-ptuning with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- PEFT
How to use bigcode/astraios-7b-ptuning with PEFT:
from peft import PeftModel from transformers import AutoModelForCausalLM base_model = AutoModelForCausalLM.from_pretrained("bigcode/starcoderbase-7b") model = PeftModel.from_pretrained(base_model, "bigcode/astraios-7b-ptuning") - Notebooks
- Google Colab
- Kaggle
File size: 445 Bytes
75d2671 | 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": "bigcode/starcoderbase-7b",
"encoder_dropout": 0.0,
"encoder_hidden_size": 4096,
"encoder_num_layers": 2,
"encoder_reparameterization_type": "MLP",
"inference_mode": true,
"num_attention_heads": 32,
"num_layers": 42,
"num_transformer_submodules": 1,
"num_virtual_tokens": 30,
"peft_type": "P_TUNING",
"revision": null,
"task_type": "CAUSAL_LM",
"token_dim": 4096
} |