i want to make pb file of the 1558M with inference

rish-16 / gpt2client

✍🏻 gpt2-client: Easy-to-use TensorFlow Wrapper for GPT-2 117M, 345M, 774M, and 1.5B Transformer Models 🤖 📝

MIT License

372 stars 74 forks source link

i want to make pb file of the 1558M with inference #18

Closed pragnakalpdev6 closed 4 years ago

pragnakalpdev6 commented 4 years ago

hi, i am trying to make pb file for the large models 1558M/774M but unable to do so. And after making PB file i want to do inference for text generation

rish-16 commented 4 years ago

Hello there, Thank you for using gpt2-client! Currently, the module does not feature a way to convert the model checkpoints to a .pb file automatically. It'd be awesome if you could create a feature request in the Issues tab with more information on what you'd like to do with it.

Cheers!