v1.0.1

docker/Dockerfile

+FROM image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-centos7.6-dtk23.10-py38
+ENV DEBIAN_FRONTEND=noninteractive
+# RUN yum update && yum install -y git cmake wget build-essential
+RUN source /opt/dtk-23.10/env.sh
+# 安装pip相关依赖
+COPY requirements.txt requirements.txt
+RUN pip3 install -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com -r requirements.txt
+

docker/requirements.txt

+#torch==2.0.1
+#torchvision==0.15.2
+#deepspeed==0.9.5
+transformers==4.33.1
+tokenizers==0.13.3
+sentencepiece==0.1.99
+shortuuid==1.0.11
+accelerate==0.21.0
+peft==0.4.0
+#bitsandbytes==0.41.0
+pydantic==1.10.13
+markdown2==2.4.8
+numpy
+scikit-learn==1.2.2
+gradio==3.35.2
+requests==2.28.2
+httpx==0.24.0
+uvicorn==0.22.0
+fastapi==0.103.0
+einops==0.6.1
+einops-exts==0.0.4
+timm==0.9.12
+#flash-attn==2.3.2