diff --git a/GPT_SoVITS/Accelerate/PyTorch/backends/torch_varlen.py b/GPT_SoVITS/Accelerate/PyTorch/backends/torch_varlen.py
index 9f687ac4..b4495cf5 100644
--- a/GPT_SoVITS/Accelerate/PyTorch/backends/torch_varlen.py
+++ b/GPT_SoVITS/Accelerate/PyTorch/backends/torch_varlen.py
@@ -134,8 +134,9 @@ class CUDAGraphCache(CUDAGraphCacheABC):
     def __init__(
         self,
         decoder,
+        cache_size: int,
     ) -> None:
-        super().__init__(decoder)
+        super().__init__(decoder, cache_size)
 
     def create_graph_cache(self, bsz: int) -> NoReturn:
         raise NotImplementedError("Cuda Graph Is Not Supported For Varlen Model")