Upload model

model.py

@@ -8,6 +8,7 @@ from scipy.ndimage import gaussian_filter1d
 from transformers import PreTrainedModel
 from in2in.utils.configs import get_config
 from in2in.models.in2in import in2IN
+from in2in.utils.preprocess import MotionNormalizer
 
 from .config import in2INConfig
 
@@ -19,6 +20,7 @@ class in2INModel(PreTrainedModel):
         super().__init__(config)
         self.mode = config.MODE
         self.model = in2IN(config, mode=config.MODE)
+        self.normalizer = MotionNormalizer()
 
     def forward(self, prompt_interaction, prompt_individual1, prompt_individual2):
         self.model.eval()

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b6a283bb5b46e53f95f2283c0a1825e9467ca0cc4d4e29c7a153a783df1f2cbf
+oid sha256:eb7e10f813dca281a85b90923d3bade36195d8cd450fcb2604c63a3d3e332af9
 size 1242367342