Skip to content
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

Any tips on speeding up the image generation? #76

Open
JaypTookMyJayp opened this issue Nov 2, 2024 · 1 comment
Open

Any tips on speeding up the image generation? #76

JaypTookMyJayp opened this issue Nov 2, 2024 · 1 comment

Comments

@JaypTookMyJayp
Copy link

my catvton takes about 12 seconds to generate an image on A100 GPU.
this is of course when the catvton's models are already loaded.
I'm using 512 * 768 image for both clothes and a model, so not a big image.

I am using fp16 precision. bf16 is bad in quality, so I can't go lower.
I did try torch.compile but it didn't help at all.
I'm using 25 steps & cfg 2.8. seems like this is the lowest I can get considering the quality.

how can I turn this into ~5 seconds process?

or at least may I get to know which process of CatVTON actually takes up the most time?

@Zheng-Chong
Copy link
Owner

It is difficult to effectively compress the model's time consumption for training-free strategy.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants