mirror of
https://github.com/RVC-Boss/GPT-SoVITS.git
synced 2025-10-15 21:26:51 +08:00
del musa_utils from musa_accelerator
This commit is contained in:
parent
50db2e9199
commit
1f91faf51e
@ -17,7 +17,6 @@ from lightning_fabric.utilities.types import _DEVICE
|
||||
from pytorch_lightning.accelerators.accelerator import Accelerator
|
||||
from pytorch_lightning.utilities.exceptions import MisconfigurationException
|
||||
|
||||
import musa_utils
|
||||
# from icecream import ic
|
||||
|
||||
import logging
|
||||
@ -55,7 +54,7 @@ class MUSAAccelerator(Accelerator):
|
||||
|
||||
@staticmethod
|
||||
def parse_devices(devices: Any) -> Any:
|
||||
return musa_utils.get_device()
|
||||
return torch.device("musa") if _musa_available else None
|
||||
|
||||
@staticmethod
|
||||
@override
|
||||
|
Loading…
x
Reference in New Issue
Block a user