Upload ./README.md with huggingface_hub
Browse files
README.md
CHANGED
|
@@ -27,7 +27,7 @@ To use the pre-trained model for text generation, use the following snippet:
|
|
| 27 |
from transformers import AutoModelForMaskedLM, AutoTokenizer
|
| 28 |
|
| 29 |
tokenizer = AutoTokenizer.from_pretrained('bert-base-uncased')
|
| 30 |
-
model = AutoModelForMaskedLM.from_pretrained('Continuous-Rivals-Discrete/langflow-
|
| 31 |
|
| 32 |
# Generate samples
|
| 33 |
samples = model.generate_samples(num_samples=5, num_steps=128)
|
|
|
|
| 27 |
from transformers import AutoModelForMaskedLM, AutoTokenizer
|
| 28 |
|
| 29 |
tokenizer = AutoTokenizer.from_pretrained('bert-base-uncased')
|
| 30 |
+
model = AutoModelForMaskedLM.from_pretrained('Continuous-Rivals-Discrete/langflow-lm1b', trust_remote_code=True)
|
| 31 |
|
| 32 |
# Generate samples
|
| 33 |
samples = model.generate_samples(num_samples=5, num_steps=128)
|