xhan77 / ssd-lm

Semi-autoregressive Simplex-based Diffusion Language Model for Text Generation and Modular Control

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Classifier-guidence generation code

yuanenming opened this issue · comments

Hi, Thanks for sharing your implementation. It is really helpful and easy to follow.

Except for the prompt-based generation, the controlled text generation is also conducted in your experiment part. I think it is called classifier guidence in the diffusion models. But I did not found the code to run the controlled text generation.

Do you have any plan to open-source this?

Sorry, I found that code in the decoding part~