This commit is contained in:
XXXXRT666 2025-10-05 08:21:56 +01:00
parent 34373c5b07
commit 4d8e5dce38

View File

@ -169,6 +169,8 @@ class T2SEngine(T2SEngineProtocol):
torch.xpu.empty_cache() torch.xpu.empty_cache()
case "mtia": case "mtia":
torch.mtia.empty_cache() torch.mtia.empty_cache()
case "cpu":
pass
match session.device.type: match session.device.type:
case "cuda": case "cuda":