FredZhang7 commited on
Commit
2f7f224
1 Parent(s): b7a7f6d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -31
README.md CHANGED
@@ -37,37 +37,7 @@ Major improvements from v1 are:
37
  pip install --upgrade transformers
38
  ```
39
 
40
- Faster but less fluent generation:
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
- ![greedy search](./greedy_search.png)
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