Requesting the inference code for sampling without using Stage 2 as shown in Figure 2 of the paper

[gaoyixuan111]

Thank you for your excellent work. I am very interested and will soon reference it. Could you please share with me the code for directly generating edited images using only the Stage 1 model, similar to the inference code for Stage 1 and Stage 2 as shown in Figure 2 of the paper.My email address is [email protected].