mirror of
https://github.com/rasbt/LLMs-from-scratch.git
synced 2026-04-10 12:33:42 +00:00
Make quote style consistent (#891)
This commit is contained in:
committed by
GitHub
parent
9276edbc37
commit
7ca7c47e4a
@@ -110,7 +110,7 @@ def test_dummy_qwen3_moe_forward(dummy_cfg_moe, dummy_input):
|
||||
out = model(dummy_input)
|
||||
assert out.shape == (1, dummy_input.size(1), dummy_cfg_moe["vocab_size"]), \
|
||||
f"Expected shape (1, seq_len, vocab_size), got {out.shape}"
|
||||
assert any(hasattr(block.ff, 'gate') for block in model.trf_blocks), \
|
||||
assert any(hasattr(block.ff, "gate") for block in model.trf_blocks), \
|
||||
"Expected MoEFeedForward in at least one transformer block"
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user