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.accelerators.accelerator import Accelerator
|
||||||
from pytorch_lightning.utilities.exceptions import MisconfigurationException
|
from pytorch_lightning.utilities.exceptions import MisconfigurationException
|
||||||
|
|
||||||
import musa_utils
|
|
||||||
# from icecream import ic
|
# from icecream import ic
|
||||||
|
|
||||||
import logging
|
import logging
|
||||||
@ -55,7 +54,7 @@ class MUSAAccelerator(Accelerator):
|
|||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def parse_devices(devices: Any) -> Any:
|
def parse_devices(devices: Any) -> Any:
|
||||||
return musa_utils.get_device()
|
return torch.device("musa") if _musa_available else None
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
@override
|
@override
|
||||||
|
Loading…
x
Reference in New Issue
Block a user