p04798526
/
LLaMA-Factory-310P3
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix torch-npu dependency

This commit is contained in:
fanjunliang 2024-06-26 18:21:42 +08:00
parent 82d0b46bc9
commit 8096f94a7d
2 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
docker/docker-npu/Dockerfile
View File

@ -7,16 +7,17 @@ ENV DEBIAN_FRONTEND=noninteractive
# Define installation arguments
ARG INSTALL_DEEPSPEED=false
ARG PIP_INDEX=https://pypi.org/simple
ARG EXTRA_INDEX=https://download.pytorch.org/whl/cpu
# Set the working directory
WORKDIR /app
# Install the requirements
COPY requirements.txt /app
RUN pip config set global.index-url $PIP_INDEX
RUN pip config set global.extra-index-url $PIP_INDEX
RUN python -m pip install --upgrade pip
RUN python -m pip install -r requirements.txt
RUN pip config set global.index-url $PIP_INDEX && \
pip config set global.extra-index-url $EXTRA_INDEX && \
pip install --upgrade pip && \
pip install -r requirements.txt
# Copy the rest of the application into the image
COPY . /app

2
setup.py
View File

@ -35,7 +35,7 @@ def get_requires():
extra_require = {
"torch": ["torch>=1.13.1"],
"torch-npu": ["torch==2.1.0", "torch-npu==2.1.0.post3", "decorator"],
"torch-npu": ["torch==2.1.0+cpu", "torch-npu==2.1.0.post3", "decorator"],
"metrics": ["nltk", "jieba", "rouge-chinese"],
"deepspeed": ["deepspeed>=0.10.0"],
"bitsandbytes": ["bitsandbytes>=0.39.0"],