From 3b9bd46b3ea6abf99fe9490d028f7324bd77ce53 Mon Sep 17 00:00:00 2001
From: XXXXRT666 <157766680+XXXXRT666@users.noreply.github.com>
Date: Fri, 24 Oct 2025 03:49:05 +0100
Subject: [PATCH] .

---
 GPT_SoVITS/Accelerate/PyTorch/backends/torch_varlen.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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")