Update README.md
Browse files
README.md
CHANGED
@@ -62,6 +62,20 @@ tokenizer = AutoTokenizer.from_pretrained("microsoft/phi-1_5", trust_remote_code
|
|
62 |
|
63 |
```
|
64 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
65 |
### Framework versions
|
66 |
|
67 |
- PEFT 0.10.0
|
|
|
62 |
|
63 |
```
|
64 |
|
65 |
+
And test
|
66 |
+
|
67 |
+
```
|
68 |
+
|
69 |
+
inputs = tokenizer([f"extract events from news.\n News: {test_data[0]['text']}"], return_tensors="pt", return_attention_mask=False)
|
70 |
+
|
71 |
+
outputs = model.generate(**inputs, max_length=512)
|
72 |
+
|
73 |
+
text = tokenizer.batch_decode(outputs)[0]
|
74 |
+
|
75 |
+
print(text)
|
76 |
+
|
77 |
+
```
|
78 |
+
|
79 |
### Framework versions
|
80 |
|
81 |
- PEFT 0.10.0
|