Update README.md

README.md

@@ -36,7 +36,7 @@ Prefix *1boy*  | [Generated *1boy* prompts](./anime_boy_settings.txt) |
 
 ![](./anime_boys.png)
 
-## Greedy Search
+## Contrastive Search
 ```python
 import torch
 from transformers import GPT2Tokenizer, GPT2LMHeadModel, pipeline
@@ -44,13 +44,13 @@ tokenizer = GPT2Tokenizer.from_pretrained('distilgpt2')
 tokenizer.add_special_tokens({'pad_token': '[PAD]'})
 model = GPT2LMHeadModel.from_pretrained('FredZhang7/anime-anything-promptgen')
 
-prompt = r'1girl, master piece,'
+prompt = r'1girl, genshin impact'
 
 # generate text using fine-tuned model
 nlp = pipeline('text-generation', model=model, tokenizer=tokenizer)
 
 # generate 10 samples using greedy search
-outs = nlp(prompt, max_length=76, num_return_sequences=10, do_sample=True, temperature=0.7, top_k=4, early_stopping=True)
+outs = nlp(prompt, max_length=76, num_return_sequences=10, do_sample=True, repetition_penalty=1.1, temperature=0.7, top_k=4, early_stopping=True)
 
 print('\nInput:\n' + 100 * '-')
 print('\033[96m' + prompt + '\033[0m')
@@ -63,7 +63,9 @@ print('\033[92m' + '\n\n'.join(outs) + '\033[0m\n')
 
 Output Example:
 
-![](./greedy_search.png)
+![](./contrastive_search.png)
+
+Please see [Fast GPT PromptGen](https://huggingface.co/FredZhang7/distilgpt2-stable-diffusion-v2) for more info on the pipeline parameters.
 
 
 ## Tips