Upload modeling_moment.py
Browse files- modeling_moment.py +1 -0
modeling_moment.py
CHANGED
|
@@ -31,6 +31,7 @@ class TimeseriesOutputs:
|
|
| 31 |
metadata: dict = None
|
| 32 |
illegal_output: bool = False
|
| 33 |
hidden_states: npt.NDArray = None # For Mists model
|
|
|
|
| 34 |
|
| 35 |
|
| 36 |
# refers: https://github.com/moment-timeseries-foundation-model/moment/blob/088b253a1138ac7e48a7efc9bf902336c9eec8d9/momentfm/utils/masking.py#L6C1-L6C2
|
|
|
|
| 31 |
metadata: dict = None
|
| 32 |
illegal_output: bool = False
|
| 33 |
hidden_states: npt.NDArray = None # For Mists model
|
| 34 |
+
input_mask_patch_view: npt.NDArray = None # For Mists model
|
| 35 |
|
| 36 |
|
| 37 |
# refers: https://github.com/moment-timeseries-foundation-model/moment/blob/088b253a1138ac7e48a7efc9bf902336c9eec8d9/momentfm/utils/masking.py#L6C1-L6C2
|