Update README.md

README.md

@@ -4,8 +4,23 @@ kakao brain에서 공개한 kogpt 6b model('kakaobrain/kogpt')을 fp16으로 저
 
 ```
 from transformers import GPTJForCausalLM
+
 model = GPTJForCausalLM.from_pretrained('kakaobrain/kogpt', cache_dir='./my_dir', revision='KoGPT6B-ryan1.5b', torch_dtype=torch.float16)
 ```
 
+### fp16 모델 로드 후 문장 생성
+
+```
+from transformers import GPTJForCausalLM, AutoTokenizer
+
+model = GPTJForCausalLM.from_pretrained('MrBananaHuman/kogpt_6b_fp16')
+model.to('cuda')
+tokenizer = AutoTokenizer.from_pretrained('MrBananaHuman/kogpt_6b_fp16')
+
+input_text = '이순신은'
+input_ids = tokenizer.encode(input_text, 
+
+```
+
 ### 참고 링크
 https://github.com/kakaobrain/kogpt/issues/6?fbclid=IwAR1KpWhuHnevQvEWV18o16k2z9TLgrXkbWTkKqzL-NDXHfDnWcIq7I4SJXM