Transformers
PyTorch
English
Chinese
bart
text2text-generation
GENIUS
conditional text generation
sketch-based text generation
data augmentation
Instructions to use beyond/genius-large with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use beyond/genius-large with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForSeq2SeqLM tokenizer = AutoTokenizer.from_pretrained("beyond/genius-large") model = AutoModelForSeq2SeqLM.from_pretrained("beyond/genius-large", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -12,7 +12,6 @@ datasets:
|
|
| 12 |
- c4
|
| 13 |
- beyond/chinese_clean_passages_80m
|
| 14 |
|
| 15 |
-
|
| 16 |
widget:
|
| 17 |
- text: "<mask> Conference on Empirical Methods <mask> submission of research papers <mask> Deep Learning <mask>"
|
| 18 |
example_title: "Example 1"
|
|
@@ -32,12 +31,12 @@ inference:
|
|
| 32 |
do_sample: True
|
| 33 |
---
|
| 34 |
|
| 35 |
-
#
|
|
|
|
| 36 |
|
| 37 |
-
**How to use:**
|
| 38 |
```
|
| 39 |
from transformers import pipeline
|
| 40 |
-
genius = pipeline("text2text-generation", model=
|
| 41 |
sketch = "<mask> Conference on Empirical Methods <mask> submission of research papers <mask> Deep Learning <mask>"
|
| 42 |
genius(sketch, num_beams=3, do_sample=True, max_length=200)[0]['generated_text']
|
| 43 |
```
|
|
|
|
| 12 |
- c4
|
| 13 |
- beyond/chinese_clean_passages_80m
|
| 14 |
|
|
|
|
| 15 |
widget:
|
| 16 |
- text: "<mask> Conference on Empirical Methods <mask> submission of research papers <mask> Deep Learning <mask>"
|
| 17 |
example_title: "Example 1"
|
|
|
|
| 31 |
do_sample: True
|
| 32 |
---
|
| 33 |
|
| 34 |
+
# GENIUS: generating text using sketches!
|
| 35 |
+
|
| 36 |
|
|
|
|
| 37 |
```
|
| 38 |
from transformers import pipeline
|
| 39 |
+
genius = pipeline("text2text-generation", model="beyond/genius-large", device=0)
|
| 40 |
sketch = "<mask> Conference on Empirical Methods <mask> submission of research papers <mask> Deep Learning <mask>"
|
| 41 |
genius(sketch, num_beams=3, do_sample=True, max_length=200)[0]['generated_text']
|
| 42 |
```
|