Update modeling_hymba.py

modeling_hymba.py

@@ -1679,7 +1679,7 @@ class HymbaBlock(nn.Module):
             A = -torch.exp(self.A_log[index].float())
 
             time_proj_bias = time_proj_bias.float() if time_proj_bias is not None else None
-            try:
+            if 1:
                 if use_precomputed_states:
                     scan_outputs = selective_state_update(
                         cache_params.ssm_states[self.layer_idx],
@@ -1714,7 +1714,8 @@ 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 Exception as e:
+            if use_precomputed_states:
+            # 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} ")