model.to(device)报错,NotImplementedError: Unsupported device type for sparse layout: xla
收藏回复举报
model.to(device)报错,NotImplementedError: Unsupported device type for sparse layout: xla
t('forum.solved') 已解决
发表于2024-10-25 15:43:52
0 查看

Traceback (most recent call last): 

  File "test.py", line 12, in <module> 

    model = whisper.load_model("turbo") 

  File "/data/jiacheng/pyProjects/speech-to-text/whisper/__init__.py", line 160, in load_model 

    return model.to(device) 

  File "/usr/local/python3.8/lib/python3.8/site-packages/torch_npu/contrib/transfer_to_npu.py", line 67, in decorated 

    return fn(*args, **kwargs) 

  File "/usr/local/python3.8/lib/python3.8/site-packages/torch_npu/utils/module.py", line 84, in to 

    return self._apply(convert) 

  File "/usr/local/python3.8/lib/python3.8/site-packages/torch/nn/modules/module.py", line 624, in _apply 

    self._buffers[key] = fn(buf) 

  File "/usr/local/python3.8/lib/python3.8/site-packages/torch_npu/utils/module.py", line 82, in convert 

    return t.to(device, dtype if t.is_floating_point() or t.is_complex() else None, non_blocking) 

  File "/usr/local/python3.8/lib/python3.8/site-packages/torch_npu/contrib/transfer_to_npu.py", line 67, in decorated 

    return fn(*args, **kwargs) 

  File "/usr/local/python3.8/lib/python3.8/site-packages/torch_npu/utils/device_guard.py", line 45, in wrapper 

    return func(*args, **kwargs) 

  File "/usr/local/python3.8/lib/python3.8/site-packages/torch_npu/utils/tensor_methods.py", line 89, in _to 

    return torch_npu._C.to(self, *args, **kwargs) 

NotImplementedError: Unsupported device type for sparse layout: xla

我要发帖子