add tensorflow version flag in setup.py

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

Closed HitLuca closed 4 years ago

HitLuca commented 4 years ago

Given that this project doesn't work with tensorflow >2.0, why isn't it indicated in the install_requiires section of setup.py?

something like


install_requires=[
...,
'tensorflow<2.0',
...
]

rish-16 commented 4 years ago

Hey,

Thanks for the headsup. Would you mind submitting a PR with this fix?

Appreciated :)