Skip to content

B 436918994#233

Merged
entrpn merged 3 commits intomainfrom
b_436918994
Aug 15, 2025
Merged

B 436918994#233
entrpn merged 3 commits intomainfrom
b_436918994

Conversation

@entrpn
Copy link
Copy Markdown
Collaborator

@entrpn entrpn commented Aug 14, 2025

  • Fix cross attention padding in splash attention
  • Add option to replicate vae.

@entrpn entrpn requested a review from coolkp August 14, 2025 22:36
coolkp
coolkp previously approved these changes Aug 14, 2025
config_fp8.quantization = "fp8"
provider_fp8 = WanPipeline.get_qt_provider(config_fp8)
self.assertIsNotNone(provider_fp8)
mock_qt_rule.assert_called_once_with(
Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

we don't want to undo this change, sanbao fixed the test earlier. There is no attributes rules so was breaking the test.

act_qtype=jnp.float8_e4m3fn,
bwd_qtype=jnp.float8_e5m2,
bwd_use_original_residuals=True,
disable_channelwise_axes=True, # per_tensor calibration
Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Ditto

@entrpn entrpn merged commit b0bd1b0 into main Aug 15, 2025
1 of 3 checks passed
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