Updated Condition Mask in Inference

Great work! I have a question regarding the mask used in the sampling process. In the sampling pseudocode in your paper, it shows that the conditional mask is updated at each timestep. However, in the code, the conditional mask appears to be fixed to the initial mask throughout all timesteps. Could you please clarify this discrepancy? Thank you!