test_generation.py

[tangleo89]

test_generation.py is not runnable.
On line 7, from continous_diffusion import DiffusionModel should be from continous_diffusion.diffusion import DiffusionModel.
On line 27-34, vocab_size is not defined in DiffusionModel, also tokenizer,p_self_cond,p_mask_cond,p_mask,prefix are not specified.

[Francesco215]

Hi, thanks for the feedback, I could fix that, but nowadays the field of discrete diffusion has gone forward, i think you should just go to look at more recent papers

if you want i can give you some references

[tangleo89]

Thanks. Agreed that discrete diffusion models are the state of the arts. I feel that diffusion models on categorical data are still of (independent) interest. It would be beneficial to others if the codes here are runnable.