nealchen commited on
Commit
d5d395d
·
verified ·
1 Parent(s): ff393be

Upload ./README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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-owt', trust_remote_code=True)
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)