Update modeling_hymba.py

modeling_hymba.py

@@ -1714,8 +1714,11 @@ class HymbaBlock(nn.Module):
     
                     if ssm_state is not None and cache_params is not None:
                         cache_params.ssm_states[self.layer_idx].copy_(ssm_state)
-            except:
-                print(f"use_precomputed_states {use_precomputed_states}; {index} {self.D}, {delta_bias} ")
+            except Exception as e:
+                print("\n\n\n\n")
+                print(e)
+                print(f"use_precomputed_states {use_precomputed_states}; {index} {self.D}, {time_proj_bias} ")
+                print("\n\n\n\n")
                     
             scan_outputs = scan_outputs.transpose(1, 2)