JSX_TTS / torch /cuda /amp /common.py
UMMJ's picture
Upload 5875 files
9dd3461
import torch
from importlib.util import find_spec
def amp_definitely_not_available():
return not (torch.cuda.is_available() or find_spec('torch_xla'))