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