Update modeling_hymba.py
Browse files- 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],
|