Skip to content

Conversation

@ziyu-guo
Copy link
Contributor

This PR adds ability to generate arbitrary-length cache schedules, on arbitrary number of calibration inputs.

The helper class implementations are under SmoothCache/calibration

Example script integrating schedule generation process into normal HuggingFace Diffusers DiT pipeline provided at examples/run_calibration.py

@kelpabc123 kelpabc123 self-assigned this Jan 24, 2025
@ziyu-guo ziyu-guo requested a review from kelpabc123 March 10, 2025 17:26
@ziyu-guo ziyu-guo merged commit 54ebe11 into main Mar 11, 2025
@ziyu-guo ziyu-guo deleted the add-schedule-gen2 branch March 11, 2025 23:03
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

Successfully merging this pull request may close these issues.

2 participants