diff --git a/seedvr2.py b/seedvr2.py
index 91fa109..d96118e 100644
--- a/seedvr2.py
+++ b/seedvr2.py
@@ -14,7 +14,7 @@
 
 import os
 import torch
-import mediapy
+#import mediapy
 from einops import rearrange
 from omegaconf import OmegaConf
 import numpy as np