mirror of
https://github.com/openai/whisper.git
synced 2025-11-24 06:26:03 +00:00
14 lines
319 B
Python
14 lines
319 B
Python
import torch
|
|
|
|
def get_x_hpu(x_numpy):
|
|
from habana_frameworks.torch.utils.library_loader import load_habana_module
|
|
|
|
load_habana_module()
|
|
|
|
x_hpu = torch.from_numpy(x_numpy).to("hpu")
|
|
return x_hpu
|
|
|
|
|
|
def is_hpu_device(device: torch.device):
|
|
return device in (torch.device("hpu:0"), torch.device("hpu"))
|