Abinaya commited on
Commit
ea91a9d
·
verified ·
1 Parent(s): 9cb453e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +14 -0
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