We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 622f721 commit 3976ea7Copy full SHA for 3976ea7
tests/test_quantization/lifecycle/test_apply.py
@@ -110,7 +110,7 @@ def test_apply_quantization_config_tinyllama():
110
# sanity check correct number of layers targeted
111
assert num_linears == 154 # 155 Linear layers - 1 that gets ignored
112
assert num_embeddings == 1
113
- assert num_rotary_embeddings == 22
+ assert num_rotary_embeddings == 23 # model updated, now has model.rotary_embedding
114
115
# test quantization compression
116
# sample forward pass to fill scales, zps
0 commit comments