google-research / xmcgan_image_generation

98 stars 15 forks source link

Error while turning self.return_text switch ON #20

Open jayati-naik opened 2 years ago

jayati-naik commented 2 years ago

Dear Team,

I want to save generated images with caption. But when I'm switching self.return_text to True. It fails at below line for conversion saying array of caption is not a valid JAX type. Please help.

https://github.com/jayati-naik/xmcgan_image_generation/blame/b03e3b5cc4885f415bc1a4b8998260412f5ae803/xmcgan/utils/eval_metrics.py#L148