@pafitis Yes that actually worked.
I changed the configuration as @jsrozner mentioned.
My settings are:
* Max_length = None
* Min_length = None
for the model.generate()
* model.config.max_length = 50
You need to change model.config.max_length also.