SKT-AI / KoGPT2

Korean GPT-2 pretrained cased (KoGPT2)
Other
532 stars 98 forks source link

feat: Add onnx convert and Update requirements.txt #31

Closed soeque1 closed 3 years ago

soeque1 commented 3 years ago

Issue #30

1. 해당 PR은 어떤 내용인가요?

ONNX export를 하고 inference 하는 내용입니다.

2. PR과 관련된 이슈가 있나요?

export가 조금 더 강조된 코드입니다. inference의 경우 휴리스틱한 코드입니다.

3. 실행

(1) kogpt2 설치 없이

PYTHONPATH=. python onnx/export_onnx_kogpt2.py 

(2) kogpt2 설치 이후

python onnx/export_onnx_kogpt2.py 
haven-jeon commented 3 years ago

README.md도 업데이트 부탁드립니다. ^^

haven-jeon commented 3 years ago

Thanks~~!