Update README.md
Browse files
README.md
CHANGED
@@ -25,7 +25,7 @@ widget:
|
|
25 |
example_title: "creative prompt"
|
26 |
parameters:
|
27 |
min_length: 16
|
28 |
-
|
29 |
no_repeat_ngram_size: 1
|
30 |
do_sample: True
|
31 |
---
|
@@ -48,7 +48,7 @@ Results in (_DALL-E, but you get the idea_):
|
|
48 |
![example](https://i.ibb.co/WkmTnZD/image.png)
|
49 |
|
50 |
|
51 |
-
|
52 |
- this `distilgpt2` version is probably small/fast enough to be used locally on CPU!
|
53 |
|
54 |
## basic usage
|
@@ -73,7 +73,8 @@ generator = pipeline(
|
|
73 |
prompt = "The Answer to Why"
|
74 |
result = generator(
|
75 |
prompt,
|
76 |
-
|
|
|
77 |
print(result[0]["generated_text"])
|
78 |
```
|
79 |
|
|
|
25 |
example_title: "creative prompt"
|
26 |
parameters:
|
27 |
min_length: 16
|
28 |
+
max_new_tokens: 24
|
29 |
no_repeat_ngram_size: 1
|
30 |
do_sample: True
|
31 |
---
|
|
|
48 |
![example](https://i.ibb.co/WkmTnZD/image.png)
|
49 |
|
50 |
|
51 |
+
<br><br>
|
52 |
- this `distilgpt2` version is probably small/fast enough to be used locally on CPU!
|
53 |
|
54 |
## basic usage
|
|
|
73 |
prompt = "The Answer to Why"
|
74 |
result = generator(
|
75 |
prompt,
|
76 |
+
max_new_tokens=24,
|
77 |
+
) # generate, adjust/add kwargs as needed
|
78 |
print(result[0]["generated_text"])
|
79 |
```
|
80 |
|