search

salesforce / GeDi

GeDi: Generative Discriminator Guided Sequence Generation
https://arxiv.org/abs/2009.06367
BSD 3-Clause "New" or "Revised" License
208 stars 47 forks source link

GEDI generate returning NONE #15

Open omerarshad opened 2 years ago

omerarshad commented 2 years ago

GEDI was successfully trained on my custom dataset, but generate function is returning nothing.

gedi_prediction_obj.generate_text("homophobia", prompt="You people are" )