423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-10-14 23:58:11 +08:00
Code Issues Packages Projects Releases Wiki Activity
LLaMA-Factory/docker/docker-npu
History
leo-pony b9f84900ee [npu] update cann base image and torch 2.4 (#7061) 
* Update base npu container image version:The Python version required for Hugging Face Transformers is >= python3.10

* Fix the bug: arg type of INSTALL_DEEPSPEED shoud been string now.

* Update Ascend CANN, CANN-Kernel and corresponding torch and torch-npu version

* Upgrade torch-npu needs packages' version: torch==2.1.0 and torch-npu==2.4.0.post2

Former-commit-id: d6dafada58412b0c801e576ef4d8d96203f792af
2025-02-25 23:32:01 +08:00
..
docker-compose.yml
[npu] update cann base image and torch 2.4 (#7061)
2025-02-25 23:32:01 +08:00
Dockerfile
[npu] update cann base image and torch 2.4 (#7061)
2025-02-25 23:32:01 +08:00