423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-27 09:10:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
hiyouga
2025-12-27 08:49:43 +08:00
parent 6cdbfa980e
commit 4d8c8b2019
2 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
docker/docker-npu/Dockerfile
View File

@@ -40,6 +40,9 @@ RUN pip uninstall -y torch torchvision torchaudio && \
# Set up volumes # Set up volumes
# VOLUME [ "/root/.cache/huggingface", "/app/shared_data", "/app/output" ] # VOLUME [ "/root/.cache/huggingface", "/app/shared_data", "/app/output" ]
# Verify torch version
RUN python -c "import torch, sys; v=torch.__version__; print('torch version:', v); sys.exit(0 if v.startswith('2.7.1') else 1)"
# Expose port 7860 for LLaMA Board # Expose port 7860 for LLaMA Board
ENV GRADIO_SERVER_PORT=7860 ENV GRADIO_SERVER_PORT=7860
EXPOSE 7860 EXPOSE 7860

1
docker/docker-npu/docker-compose.yml
View File

@@ -35,7 +35,6 @@ services:
args: args:
BASE_IMAGE: quay.io/ascend/cann:8.3.rc2-a3-ubuntu22.04-py3.11 BASE_IMAGE: quay.io/ascend/cann:8.3.rc2-a3-ubuntu22.04-py3.11
PIP_INDEX: https://pypi.org/simple PIP_INDEX: https://pypi.org/simple
EXTRAS: torch-npu,metrics
container_name: llamafactory-a3 container_name: llamafactory-a3
image: llamafactory:npu-a3 image: llamafactory:npu-a3
volumes: volumes: