zpbrent commited on
Commit
ac58b46
1 Parent(s): e4c4db3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -1
README.md CHANGED
@@ -117,11 +117,14 @@ The authors provide additionally notes about the training procedure used, in the
117
  ```
118
  from transformers import TransfoXLTokenizer, TransfoXLModel
119
  import torch
 
120
  tokenizer = TransfoXLTokenizer.from_pretrained("zpbrent/newInfect")
121
  model = TransfoXLModel.from_pretrained("zpbrent/newInfect")
 
122
  inputs = tokenizer("Hello, my dog is cute", return_tensors="pt")
123
  outputs = model(**inputs)
124
- last_hidden_states = outputs.last_hidden_state
 
125
  ```
126
 
127
 
 
117
  ```
118
  from transformers import TransfoXLTokenizer, TransfoXLModel
119
  import torch
120
+
121
  tokenizer = TransfoXLTokenizer.from_pretrained("zpbrent/newInfect")
122
  model = TransfoXLModel.from_pretrained("zpbrent/newInfect")
123
+
124
  inputs = tokenizer("Hello, my dog is cute", return_tensors="pt")
125
  outputs = model(**inputs)
126
+
127
+ print(outputs)
128
  ```
129
 
130