search

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

Unable to specify length of output? #23

Open Emperornero opened 4 years ago

Emperornero commented 4 years ago

Not much to be said.

When attempting to call the length parameter of gpt-2 generate function, the functioncall fails with "TypeError: generate() got an unexpected keyword argument 'length'"

Is changing the length of the output not supported?

rish-16 commented 4 years ago

Hello there, I apologise for the sudden errors. I'm in the process of writing a patch over the generate function to allow for more flexibility when generating text. It should be out in the next release. I appreciate your patience!

callmetushar123 commented 3 years ago

@rish-16 Hello Did you finish adding that feature? or maybe just put that code as gist. Will be helpful for me.