HachiML
/

MOMENT-1-large-embedding-v0.1

Feature Extraction

Model card Files Files and versions Community

HachiML commited on Jun 26

Commit

1d6c599

•

1 Parent(s): 352385f

Upload modeling_moment.py

Files changed (1) hide show

modeling_moment.py +1 -0

modeling_moment.py CHANGED Viewed

@@ -31,6 +31,7 @@ class TimeseriesOutputs:
     metadata: dict = None
     illegal_output: bool = False
     hidden_states: npt.NDArray = None  # For Mists model
 # refers: https://github.com/moment-timeseries-foundation-model/moment/blob/088b253a1138ac7e48a7efc9bf902336c9eec8d9/momentfm/utils/masking.py#L6C1-L6C2

     metadata: dict = None
     illegal_output: bool = False
     hidden_states: npt.NDArray = None  # For Mists model
+    input_mask_patch_view: npt.NDArray = None  # For Mists model
 # refers: https://github.com/moment-timeseries-foundation-model/moment/blob/088b253a1138ac7e48a7efc9bf902336c9eec8d9/momentfm/utils/masking.py#L6C1-L6C2