Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • myhloli's avatar
    fix(npu): correct module name for NPU operations · 2684e775
    myhloli authored 
    - Update `clean_memory.py` to use `torch_npu.npu` instead of `torch.npu`
- Update `model_utils.py` to use `torch_npu.npu` instead of `torch.npu`
- Simplify NPU availability check and bfloat16 support in `pdf_parse_union_core_v2.py`
    2684e775
model_utils.py 2.43 KB