[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2026-07-03 14:55:53 +00:00
commit 11a00d0238

View file

@ -245,7 +245,10 @@ def test_krea2_forward_matches_stock():
torch.manual_seed(4)
blk = Krea2TransformerBlock(
hidden_size = D, intermediate_size = 2 * D, num_heads = H, num_kv_heads = H // 2,
hidden_size = D,
intermediate_size = 2 * D,
num_heads = H,
num_kv_heads = H // 2,
norm_eps = 1e-6,
).eval()
# Give the zero-init modulation table real values so all six scale/shift/gate