Update README.md

README.md

@@ -26,7 +26,9 @@ You can use this model directly with a pipeline for text generation. Since the g
 
 ```python
 from transformers import pipeline, set_seed
-generator = pipeline('text-generation', model='nlp-waseda/gpt2-xl-japanese')
+generator = pipeline('text-generation', model='nlp-waseda/gpt2-xl-japanese
+# if you have a gpu.
+# generator = pipeline('text-generation', model='nlp-waseda/gpt2-xl-japanese', device=0)
 
 set_seed(42)
 generator("早稲田 大学 で 自然 言語 処理 を", max_length=30, do_sample=True, pad_token_id=2, num_return_sequences=5)