423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2026-06-17 04:38:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

[model] Patch GDN for NPU (#10504)

Browse Source 
Co-authored-by: jiaqiw09 <jiaqiw960714@gmail.com>
This commit is contained in:
A1waysBeenHere
2026-06-04 16:39:02 +08:00
committed by GitHub
parent 053d43c0ac
commit 409e8a477f
14 changed files with 3219 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
docker/docker-npu/Dockerfile
View File

@@ -36,6 +36,7 @@ COPY . /app
RUN source /usr/local/Ascend/ascend-toolkit/set_env.sh
RUN pip uninstall -y torch torchvision torchaudio
RUN pip install --no-cache-dir -r requirements/npu.txt --index-url "${PYTORCH_INDEX}"
RUN pip install --no-cache-dir -r requirements/triton_ascend.txt
RUN pip install --no-cache-dir -r requirements/deepspeed.txt
RUN pip install --no-cache-dir -e . --no-build-isolation && \
pip install --no-cache-dir -r requirements/metrics.txt --no-build-isolation