Spaces:

pineconeT94
/

fishspeech

Running

fishspeech / fish_speech /utils /context.py

deploy

20528a2 almost 2 years ago

287 Bytes

from contextlib import nullcontext

import torch


def autocast_exclude_mps(
    device_type: str, dtype: torch.dtype
) -> nullcontext | torch.autocast:
    return (
        nullcontext()
        if torch.backends.mps.is_available()
        else torch.autocast(device_type, dtype)
    )

	from contextlib import nullcontext

	import torch


	def autocast_exclude_mps(
	device_type: str, dtype: torch.dtype
	) -> nullcontext \| torch.autocast:
	return (
	nullcontext()
	if torch.backends.mps.is_available()
	else torch.autocast(device_type, dtype)
	)