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
|
@@ -44,12 +44,10 @@ inference:
|
|
| 44 |
You can use this model directly with a pipeline for masked language modeling:
|
| 45 |
|
| 46 |
```python
|
| 47 |
-
from transformers import
|
| 48 |
-
|
| 49 |
-
|
| 50 |
-
|
| 51 |
-
|
| 52 |
-
|
| 53 |
-
generated_text = genius(sketch, num_beams=3, do_sample=True, max_length=200)[0]['generated_text']
|
| 54 |
-
print(generated_text)
|
| 55 |
```
|
|
|
|
| 44 |
You can use this model directly with a pipeline for masked language modeling:
|
| 45 |
|
| 46 |
```python
|
| 47 |
+
from transformers import BertTokenizer, TFBertModel
|
| 48 |
+
tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
|
| 49 |
+
model = TFBertModel.from_pretrained("bert-base-uncased")
|
| 50 |
+
text = "Replace me by any text you'd like."
|
| 51 |
+
encoded_input = tokenizer(text, return_tensors='tf')
|
| 52 |
+
output = model(encoded_input)
|
|
|
|
|
|
|
| 53 |
```
|