Q&A function in your code

[Wiselnn570]

Hi, @mariannearr! Could you please provide a clean Q&A function in your code?

I think it’s hard to follow—especially for those who are not familiar with diffusion models, since there are so many hyperparameters involved and all the training and validation processes are wrapped inside a single main.py. I've noticed that others in the community share the same confusion(see #14 (comment)).

I hope you might consider adding this part to the code, which I believe would promote more follow-up for this work.

[mariannearr]

Hello, I will add Q&A functionality to BD3LMs to facilitate future research. I am currently traveling for ICLR, and will add this implementation after I return.

NOTE: our small models trained on OWT are expected to perform poorly on Q&A tasks, similar to the results from GPT2-small