pszemraj commited on
Commit
4fbe0a7
1 Parent(s): fbb0f7d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +14 -0
README.md CHANGED
@@ -22,6 +22,20 @@ It achieves the following results on the evaluation set:
22
  - Gen Len: 12.0586
23
  - Num Input Tokens Seen: 1931815668
24
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
25
  ## Quick eval
26
 
27
  Quick eval for: `BEE-spoke-data/tFINE-900m-e16-d32-flan`
 
22
  - Gen Len: 12.0586
23
  - Num Input Tokens Seen: 1931815668
24
 
25
+ ## Usage Example
26
+
27
+ ```py
28
+ from transformers import pipeline
29
+
30
+ pipe = pipeline(
31
+ "text2text-generation",
32
+ model="BEE-spoke-data/tFINE-900m-e16-d32-flan",
33
+ )
34
+ prompt = "What color is tuesday?"
35
+ res = pipe(prompt, max_new_tokens=96, top_k=4, penalty_alpha=0.6)
36
+ print(res[0]["generated_text"])
37
+ ```
38
+
39
  ## Quick eval
40
 
41
  Quick eval for: `BEE-spoke-data/tFINE-900m-e16-d32-flan`