loubnabnl HF staff commited on
Commit
4fa0d0f
1 Parent(s): 4bd868a
Files changed (1) hide show
  1. architectures/codeparrot.txt +23 -1
architectures/codeparrot.txt CHANGED
@@ -3,4 +3,26 @@
3
  |Model | # parameters |
4
  | - | - |
5
  | GPT2 | 110M |
6
- | GPT2 | 1.5B |
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
3
  |Model | # parameters |
4
  | - | - |
5
  | GPT2 | 110M |
6
+ | GPT2 | 1.5B |
7
+
8
+ You can load the model and tokenizer directly from the `transformers`:
9
+
10
+ ```python
11
+ from transformers import AutoTokenizer, AutoModelWithLMHead
12
+
13
+ tokenizer = AutoTokenizer.from_pretrained("lvwerra/codeparrot")
14
+ model = AutoModelWithLMHead.from_pretrained("lvwerra/codeparrot")
15
+
16
+ inputs = tokenizer("def hello_world():", return_tensors="pt")
17
+ outputs = model(**inputs)
18
+
19
+ ```
20
+
21
+ Or you can use a pipeline
22
+
23
+ ```python
24
+ from transformers import pipeline
25
+
26
+ pipe = pipeline("text-generation", model="lvwerra/codeparrot")
27
+ outputs = pipe("def hello_world():")
28
+ ```