Skip to content

Patch WanTimeTextImageEmbedding forward only with fp8#327

Merged
a-r-r-o-w merged 1 commit into
mainfrom
fix/wan-fp8-patch
Mar 15, 2025
Merged

Patch WanTimeTextImageEmbedding forward only with fp8#327
a-r-r-o-w merged 1 commit into
mainfrom
fix/wan-fp8-patch

Conversation

@a-r-r-o-w
Copy link
Copy Markdown
Contributor

@a-r-r-o-w a-r-r-o-w commented Mar 15, 2025

#316 fixed a bug for FP8 layerwise casting training. However, the fix breaks bf16 training. This PR enables the patch only if using FP8 casting

@a-r-r-o-w a-r-r-o-w merged commit 3dcb466 into main Mar 15, 2025
@a-r-r-o-w a-r-r-o-w deleted the fix/wan-fp8-patch branch March 15, 2025 08:29
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.

1 participant