Create README.md
Browse files
README.md
ADDED
@@ -0,0 +1,11 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
```python
|
2 |
+
from transformers import GPTNeoXForCausalLM, AutoTokenizer
|
3 |
+
|
4 |
+
tokenizer = AutoTokenizer.from_pretrained("afterless/reverse-pythia-160m")
|
5 |
+
model = GPTNeoXForCausalLM.from_pretrained("afterless/reverse-pythia-160m")
|
6 |
+
|
7 |
+
inputs = tokenizer("but I told him, the cheese was the best", return_token_type_ids=False, return_tensors="pt")
|
8 |
+
inputs['input_ids'] = t.flip(inputs.input_ids, (1,))
|
9 |
+
tokens = t.flip(model.generate(**inputs), (1,))
|
10 |
+
tokenizer.decode(tokens[0])
|
11 |
+
```
|