Instructions to use pruhtopia/bart-toc-generation with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use pruhtopia/bart-toc-generation with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForSeq2SeqLM tokenizer = AutoTokenizer.from_pretrained("pruhtopia/bart-toc-generation") model = AutoModelForSeq2SeqLM.from_pretrained("pruhtopia/bart-toc-generation", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Download checkpoint-360/generation_config.json from pruhtopia/bart-toc-generation: direct link, hf CLI and curl.
- Browser
- Download file 350 Bytes
-
https://huggingface.co/pruhtopia/bart-toc-generation/resolve/main/checkpoint-360/generation_config.json
- Command line
-
hf download hf://pruhtopia/bart-toc-generation/checkpoint-360/generation_config.json
-
curl -L -o generation_config.json https://huggingface.co/pruhtopia/bart-toc-generation/resolve/main/checkpoint-360/generation_config.json
350 Bytes
| { | |
| "bos_token_id": 0, | |
| "decoder_start_token_id": 2, | |
| "early_stopping": true, | |
| "eos_token_id": 2, | |
| "forced_bos_token_id": 0, | |
| "forced_eos_token_id": 2, | |
| "length_penalty": 2.0, | |
| "max_length": 142, | |
| "min_length": 56, | |
| "no_repeat_ngram_size": 3, | |
| "num_beams": 4, | |
| "pad_token_id": 1, | |
| "transformers_version": "4.37.0", | |
| "use_cache": false | |
| } | |