blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2026-06-30 02:08:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

.

Browse Source
This commit is contained in:
XXXXRT666 2025-10-24 04:20:15 +01:00
parent f25fcfb0e4
commit 0651cb5d01
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
GPT_SoVITS/Accelerate/PyTorch/t2s_model_abc.py
View File

@ -692,6 +692,7 @@ class CUDAGraphCacheABC(ABC):
self.graph_cache: dict[int, Queue[CUDAGraphStateABC]] = {}
if torch.cuda.is_available() and torch.version.cuda is not None and os.environ.get("CUDAGraph", "1") != "0":
self.graph_cache[1] = Queue()
self.create_graph_cache(1)
def __getitem__(self, bsz: int) -> CUDAGraphStateABC: