Code

Adding support for Ascend NPU#372

Code

Merged

Narsil merged 4 commits intohuggingface:mainhuggingface/safetensors:mainfrom ji-huazhong:support-npuji-huazhong/safetensors:support-npuCopy head branch name to clipboardNov 20, 2023

+35

Commits

Commits on Nov 1, 2023

• Adding support for Ascend NPU

  ji-huazhong

  committed
  42f94cf

  Copy full SHA for 42f94cf

Commits on Nov 18, 2023

• remove the unnecessary hack code

  ji-huazhong

  committed
  4cc6db2

  Copy full SHA for 4cc6db2

• test more dtype

  ji-huazhong

  committed
  38c761c

  Copy full SHA for 38c761c

• npu doesn't support calling torch.allclose with bf16 for now

  ji-huazhong

  committed
  cd6c531

  Copy full SHA for cd6c531