New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Very nice work, About in-between? #8

Open

lxyxl0216 opened this issue Jan 6, 2025 · 1 comment

lxyxl0216 commented Jan 6, 2025

Does the model support editing? included in-between

lxyxl0216 changed the title ~~About in-between?~~ Very nice work, About in-between?

Owner

exitudio commented Jan 6, 2025

I didn’t clean up the code for editing generation, but the code is already in the evaluation function. Here are some information:

Change edit_task to one of these: "inpainting, outpainting, prefix, suffix"
It will encode the condition tokens and replace [Mask] tokens where to inpaint here
Then it will call the edit function. The model will automatically create masked attention based on the tokens (with [Mask]) and generate tokens autoregressively with condition.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment