Instructions to use PSW/samsum_reverse_train with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use PSW/samsum_reverse_train with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForSeq2SeqLM tokenizer = AutoTokenizer.from_pretrained("PSW/samsum_reverse_train") model = AutoModelForSeq2SeqLM.from_pretrained("PSW/samsum_reverse_train", device_map="auto") - Notebooks
- Google Colab
- Kaggle
add model
Browse files- config.json +1 -0
config.json
CHANGED
|
@@ -39,6 +39,7 @@
|
|
| 39 |
"LABEL_1": 1,
|
| 40 |
"LABEL_2": 2
|
| 41 |
},
|
|
|
|
| 42 |
"max_position_embeddings": 1024,
|
| 43 |
"model_type": "bart",
|
| 44 |
"no_repeat_ngram_size": 3,
|
|
|
|
| 39 |
"LABEL_1": 1,
|
| 40 |
"LABEL_2": 2
|
| 41 |
},
|
| 42 |
+
"max_length": 200,
|
| 43 |
"max_position_embeddings": 1024,
|
| 44 |
"model_type": "bart",
|
| 45 |
"no_repeat_ngram_size": 3,
|