diff --git a/benchmark/ATD12K.py b/benchmark/ATD12K.py
index 0d8be5e..adf17a6 100644
--- a/benchmark/ATD12K.py
+++ b/benchmark/ATD12K.py
@@ -7,7 +7,7 @@ import torch
 import argparse
 import numpy as np
 from torch.nn import functional as F
-from pytorch_msssim import ssim_matlab
+from model.pytorch_msssim import ssim_matlab
 from model.RIFE import Model
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
diff --git a/benchmark/HD.py b/benchmark/HD.py
index f91153d..981f07e 100644
--- a/benchmark/HD.py
+++ b/benchmark/HD.py
@@ -7,7 +7,7 @@ import torch
 import argparse
 import numpy as np
 from torch.nn import functional as F
-from pytorch_msssim import ssim_matlab
+from model.pytorch_msssim import ssim_matlab
 from model.RIFE import Model
 from skimage.color import rgb2yuv, yuv2rgb
 from yuv_frame_io import YUV_Read,YUV_Write
diff --git a/benchmark/HD_multi.py b/benchmark/HD_multi.py
index 988ddc7..3ff1ffe 100644
--- a/benchmark/HD_multi.py
+++ b/benchmark/HD_multi.py
@@ -7,7 +7,7 @@ import torch
 import argparse
 import numpy as np
 from torch.nn import functional as F
-from pytorch_msssim import ssim_matlab
+from model.pytorch_msssim import ssim_matlab
 from model.RIFE import Model
 from skimage.color import rgb2yuv, yuv2rgb
 from yuv_frame_io import YUV_Read,YUV_Write
diff --git a/benchmark/MiddleBury_Other.py b/benchmark/MiddleBury_Other.py
index a6676ce..d0683cf 100644
--- a/benchmark/MiddleBury_Other.py
+++ b/benchmark/MiddleBury_Other.py
@@ -7,7 +7,7 @@ import torch
 import argparse
 import numpy as np
 from torch.nn import functional as F
-from pytorch_msssim import ssim_matlab
+from model.pytorch_msssim import ssim_matlab
 from model.RIFE import Model
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
diff --git a/benchmark/UCF101.py b/benchmark/UCF101.py
index 1ea20f8..9f2c424 100644
--- a/benchmark/UCF101.py
+++ b/benchmark/UCF101.py
@@ -7,7 +7,7 @@ import torch
 import argparse
 import numpy as np
 from torch.nn import functional as F
-from pytorch_msssim import ssim_matlab
+from model.pytorch_msssim import ssim_matlab
 from model.RIFE import Model
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
diff --git a/benchmark/Vimeo90K.py b/benchmark/Vimeo90K.py
index ccc7d28..977e137 100644
--- a/benchmark/Vimeo90K.py
+++ b/benchmark/Vimeo90K.py
@@ -7,8 +7,7 @@ import torch
 import argparse
 import numpy as np
 from torch.nn import functional as F
-from pytorch_msssim import ssim_matlab
-# from model.RIFE2F15C import Model
+from model.pytorch_msssim import ssim_matlab
 from model.RIFE import Model
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
diff --git a/inference_video.py b/inference_video.py
index e12b38d..57bc21c 100644
--- a/inference_video.py
+++ b/inference_video.py
@@ -9,7 +9,7 @@ import warnings
 import _thread
 import skvideo.io
 from queue import Queue, Empty
-from benchmark.pytorch_msssim import ssim_matlab
+from model.pytorch_msssim import ssim_matlab
 
 warnings.filterwarnings("ignore")
 
diff --git a/benchmark/pytorch_msssim/__init__.py b/model/pytorch_msssim/__init__.py
similarity index 100%
rename from benchmark/pytorch_msssim/__init__.py
rename to model/pytorch_msssim/__init__.py