pmolchanov commited on
Commit
6664f62
·
verified ·
1 Parent(s): a26bc5b

Update modeling_hymba.py

Browse files
Files changed (1) hide show
  1. modeling_hymba.py +1 -1
modeling_hymba.py CHANGED
@@ -1717,7 +1717,7 @@ class HymbaBlock(nn.Module):
1717
  if use_precomputed_states:
1718
  # except Exception as e:
1719
  print("\n\n\n\n")
1720
- print(e)
1721
  print(f"use_precomputed_states {use_precomputed_states}; {index} {self.D}, {time_proj_bias} ")
1722
  print(f"{self.D[index]} ")
1723
  # cache_params.ssm_states[self.layer_idx],
 
1717
  if use_precomputed_states:
1718
  # except Exception as e:
1719
  print("\n\n\n\n")
1720
+ # print(e)
1721
  print(f"use_precomputed_states {use_precomputed_states}; {index} {self.D}, {time_proj_bias} ")
1722
  print(f"{self.D[index]} ")
1723
  # cache_params.ssm_states[self.layer_idx],