Spaces:

dangtiendatdat
/

real

Runtime error

13ba381 almost 2 years ago

272 Bytes

import logging

import torch


def get_device_type() -> str:
    if torch.backends.mps.is_available():
        return "mps"
    elif torch.cuda.is_available():
        return "cuda"
    else:
        logging.warning("No GPU found, using CPU instead")
        return "cpu"

	import logging

	import torch


	def get_device_type() -> str:
	if torch.backends.mps.is_available():
	return "mps"
	elif torch.cuda.is_available():
	return "cuda"
	else:
	logging.warning("No GPU found, using CPU instead")
	return "cpu"