keras gpt 2:加载GPT 2检查点并生成文本 源码

凯拉斯GPT-2 [| ] 加载预训练的权重并使用预测。 安装 pip install keras-gpt-2 演示版 import os from keras_gpt_2 import load_trained_model_from_checkpoint , get_bpe_from_files , generate model_folder = 'xxx/yyy/117M' config_path = os . path . join ( model_folder , 'hparams.json' ) checkpoint_path = os . path . join ( model