Traceback (most recent call last):
  File "<frozen runpy>", line 198, in _run_module_as_main
  File "<frozen runpy>", line 88, in _run_code
  File "/opt/exp-model-parameter-similarity/functional/reference_check.py", line 142, in <module>
    if __name__=='__main__':main()
                            ~~~~^^
  File "/opt/exp-model-parameter-similarity/functional/reference_check.py", line 102, in main
    tout=layer.linear_attn(tn)
  File "/opt/exp-model-parameter-similarity/venv/lib/python3.14/site-packages/torch/nn/modules/module.py", line 1787, in _wrapped_call_impl
    return self._call_impl(*args, **kwargs)
           ~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^
  File "/opt/exp-model-parameter-similarity/venv/lib/python3.14/site-packages/torch/nn/modules/module.py", line 1798, in _call_impl
    return forward_call(*args, **kwargs)
  File "pinned_transformers_reference", line 542, in forward
NameError: name 'apply_mask_to_padding_states' is not defined