Skip to content

Conversation

DefTruth
Copy link
Contributor

@DefTruth DefTruth commented Sep 2, 2025

I'm the maintainer of cache-dit. cache-dit relies on type annotations to match the forward pattern of blocks more precisely. However, I found that there are a large number of type annotation errors in diffusers. For instance, HiDream, HunyuanVideo, cogvideox, auraflow, cogview, lumina, etc. This PR has fixed all related errors.

PTAL ~ @a-r-r-o-w @sayakpaul

for example, fix hidream type hint:

torch.Tensor -> Union[torch.Tensor, Tuple[torch.Tensor, torch.Tensor]]

@DefTruth DefTruth changed the title fix hidream type hint fix many type hint errors Sep 5, 2025
@DefTruth DefTruth closed this Sep 5, 2025
@DefTruth DefTruth deleted the fix-hidream-type-hint branch September 5, 2025 05:38
@sayakpaul
Copy link
Member

Why close it?

@DefTruth
Copy link
Contributor Author

DefTruth commented Sep 5, 2025

Why close it?

i have open a new pr: #12289

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