blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-10-16 05:36:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

clean

Browse Source
This commit is contained in:
KakaruHayate 2025-10-13 22:21:50 +08:00
parent fd8c860f49
commit 90141d2029
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
musa_accelerator.py
View File

@ -36,10 +36,10 @@ class MUSAAccelerator(Accelerator):
device = torch.device(f"musa:0")
torch.musa.set_device(device)
@override
def get_device_stats(self, device: _DEVICE) -> dict[str, Any]:
if device.type == "musa":
return torch_musa.memory_stats(device)
#@override
#def get_device_stats(self, device: _DEVICE) -> dict[str, Any]:
# if device.type == "musa":
# return torch_musa.memory_stats(device)
@override
def teardown(self) -> None: