Spaces:

bilegentile
/

test

Runtime error

Upload folder using huggingface_hub

c19ca42 verified over 2 years ago

248 Bytes

from typing import Optional, Union
import torch


rDevice = Union[torch.device, int]
def get_device(device: Optional[rDevice]=None) -> torch.device:
    if device is None:
        device = torch.dml.current_device()
    return torch.device(device)

	from typing import Optional, Union
	import torch


	rDevice = Union[torch.device, int]
	def get_device(device: Optional[rDevice]=None) -> torch.device:
	if device is None:
	device = torch.dml.current_device()
	return torch.device(device)