IG error
Error when calculating IG:
READY
IG failed: cannot unpack non-iterable NoneType object Traceback (most recent call last): File "/app/app.py", line 869, in main ig_a,ig_b = compute_ig(st.session_state.model,seq_a,seq_b,steps=15) ~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/app/app.py", line 535, in compute_ig attr_a = riemann(emb_a, baseline_a, emb_b, False) File "/app/app.py", line 525, in riemann out = fwd(fixed, interp) if tgt_is_b else fwd(interp, fixed) ~~~^^^^^^^^^^^^^^^ File "/app/app.py", line 519, in fwd return model.projection_head(encode(e_a, mask_a), encode(e_b, mask_b)) ~~~~~~^^^^^^^^^^^^^ File "/app/app.py", line 512, in encode h = esm.base_model.model.encoder( ~~~~~~~~~~~~~~~~~~~~~~~~~~~~^ embs.float(), attention_mask=ext ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ ).last_hidden_state.float() ^ File "/usr/local/lib/python3.13/site-packages/torch/nn/modules/module.py", line 1779, in _wrapped_call_impl return self._call_impl(*args, **kwargs) ~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^ File "/usr/local/lib/python3.13/site-packages/torch/nn/modules/module.py", line 1790, in _call_impl return forward_call(*args, **kwargs) File "/usr/local/lib/python3.13/site-packages/transformers/utils/generic.py", line 900, in wrapper output = func(self, *args, **kwargs) File "/usr/local/lib/python3.13/site-packages/transformers/models/esm/modeling_esm.py", line 556, in forward hidden_states = layer_module( hidden_states, ...<4 lines>... **kwargs, ) File "/usr/local/lib/python3.13/site-packages/transformers/modeling_layers.py", line 93, in call return super().call(*args, **kwargs) ~~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^ File "/usr/local/lib/python3.13/site-packages/torch/nn/modules/module.py", line 1779, in _wrapped_call_impl return self._call_impl(*args, **kwargs) ~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^ File "/usr/local/lib/python3.13/site-packages/torch/nn/modules/module.py", line 1790, in _call_impl return forward_call(*args, **kwargs) File "/usr/local/lib/python3.13/site-packages/transformers/models/esm/modeling_esm.py", line 504, in forward attention_output = self.attention( hidden_states, ...<2 lines>... **kwargs, ) File "/usr/local/lib/python3.13/site-packages/torch/nn/modules/module.py", line 1779, in _wrapped_call_impl return self._call_impl(*args, **kwargs) ~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^ File "/usr/local/lib/python3.13/site-packages/torch/nn/modules/module.py", line 1790, in _call_impl return forward_call(*args, **kwargs) File "/usr/local/lib/python3.13/site-packages/transformers/models/esm/modeling_esm.py", line 443, in forward attn_output, _ = self.self( ~~~~~~~~~^ hidden_states_ln, ^^^^^^^^^^^^^^^^^ ...<4 lines>... **kwargs, ^^^^^^^^^ ) ^ File "/usr/local/lib/python3.13/site-packages/torch/nn/modules/module.py", line 1779, in _wrapped_call_impl return self._call_impl(*args, **kwargs) ~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^ File "/usr/local/lib/python3.13/site-packages/torch/nn/modules/module.py", line 1790, in _call_impl return forward_call(*args, **kwargs) File "/usr/local/lib/python3.13/site-packages/transformers/models/esm/modeling_esm.py", line 390, in forward cos, sin = position_embeddings ^^^^^^^^ TypeError: cannot unpack non-iterable NoneType object