Commit
·
2f7f224
1
Parent(s):
b7a7f6d
Update README.md
Browse files
README.md
CHANGED
@@ -37,37 +37,7 @@ Major improvements from v1 are:
|
|
37 |
pip install --upgrade transformers
|
38 |
```
|
39 |
|
40 |
-
|
41 |
-
|
42 |
-
```python
|
43 |
-
from transformers import GPT2Tokenizer, GPT2LMHeadModel
|
44 |
-
tokenizer = GPT2Tokenizer.from_pretrained('distilgpt2')
|
45 |
-
tokenizer.add_special_tokens({'pad_token': '[PAD]'})
|
46 |
-
model = GPT2LMHeadModel.from_pretrained('FredZhang7/distilgpt2-stable-diffusion-v2', pad_token_id=tokenizer.eos_token_id)
|
47 |
-
|
48 |
-
prompt = r'a cat sitting'
|
49 |
-
|
50 |
-
# generate text using fine-tuned model
|
51 |
-
from transformers import pipeline
|
52 |
-
nlp = pipeline('text-generation', model=model, tokenizer=tokenizer)
|
53 |
-
|
54 |
-
# generate 5 samples
|
55 |
-
outs = nlp(prompt, max_length=80, num_return_sequences=5)
|
56 |
-
|
57 |
-
print('\nInput:\n' + 100 * '-')
|
58 |
-
print('\033[96m' + prompt + '\033[0m')
|
59 |
-
print('\nOutput:\n' + 100 * '-')
|
60 |
-
for i in range(len(outs)):
|
61 |
-
outs[i] = str(outs[i]['generated_text']).replace(' ', '')
|
62 |
-
print('\033[92m' + '\n\n'.join(outs) + '\033[0m\n')
|
63 |
-
```
|
64 |
-
|
65 |
-
Example output:
|
66 |
-

|
67 |
-
|
68 |
-
<br>
|
69 |
-
|
70 |
-
Slower but more fluent generation:
|
71 |
|
72 |
```python
|
73 |
from transformers import GPT2Tokenizer, GPT2LMHeadModel
|
|
|
37 |
pip install --upgrade transformers
|
38 |
```
|
39 |
|
40 |
+
Fast, fluent generation:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
41 |
|
42 |
```python
|
43 |
from transformers import GPT2Tokenizer, GPT2LMHeadModel
|