Instructions to use andstor/bigcode-starcoder2-15b-unit-test-prompt-tuning with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- PEFT
How to use andstor/bigcode-starcoder2-15b-unit-test-prompt-tuning with PEFT:
from peft import PeftModel from transformers import AutoModelForCausalLM base_model = AutoModelForCausalLM.from_pretrained("bigcode/starcoder2-15b") model = PeftModel.from_pretrained(base_model, "andstor/bigcode-starcoder2-15b-unit-test-prompt-tuning") - Notebooks
- Google Colab
- Kaggle
Download adapter_config.json from andstor/bigcode-starcoder2-15b-unit-test-prompt-tuning: direct link, hf CLI and curl.
- Browser
- Download file 471 Bytes
-
https://huggingface.co/andstor/bigcode-starcoder2-15b-unit-test-prompt-tuning/resolve/main/adapter_config.json
- Command line
-
hf download hf://andstor/bigcode-starcoder2-15b-unit-test-prompt-tuning/adapter_config.json
-
curl -L -o adapter_config.json https://huggingface.co/andstor/bigcode-starcoder2-15b-unit-test-prompt-tuning/resolve/main/adapter_config.json
471 Bytes
| { | |
| "auto_mapping": null, | |
| "base_model_name_or_path": "bigcode/starcoder2-15b", | |
| "inference_mode": true, | |
| "num_attention_heads": 48, | |
| "num_layers": 40, | |
| "num_transformer_submodules": 1, | |
| "num_virtual_tokens": 20, | |
| "peft_type": "PROMPT_TUNING", | |
| "prompt_tuning_init": "RANDOM", | |
| "prompt_tuning_init_text": null, | |
| "revision": null, | |
| "task_type": "CAUSAL_LM", | |
| "token_dim": 6144, | |
| "tokenizer_kwargs": null, | |
| "tokenizer_name_or_path": "bigcode/starcoder2-15b" | |
| } |