-
Notifications
You must be signed in to change notification settings - Fork 1
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.
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
Does this apply to unet? #2
Comments
Hi there, thank you for your interest in this work. We're doing additional experiments to measure the effectiveness of this technique on UNet architecture. Will keep you posted. |
Thanks for your reply❤️, I tried on sdxl and there are two problems(used threshold_0.08 json).
|
Hi, the schedules only currently work on the DiT model, as while the technique works on multiple models, you need to run an initial calibration pass to generate the schedule for each architecture (as different checkpoints have different optimal caching schedules). To generate one that works for SDXL, you will need to regenerate the schedule specifically for that architecture. Stay tuned for sample code that will enable that. If you can't wait, you can also take a stab at generating the schedule yourself with SDXL based on what we described in our paper. |
No description provided.
The text was updated successfully, but these errors were encountered: