添加启动脚本和 HF_HOME 环境变量

2024-01-26 11:37:13 +08:00
parent 766ac8704a
commit 773eb8a1bf
3 changed files with 11 additions and 2 deletions
--- a/4
+++ b/4
@@ -6,8 +6,10 @@ WORKDIR /app

 COPY ./requirements_version.txt .

-RUN pip install -r requirements_version.txt --no-cache-dir
+RUN pip install -r requirements_version.txt --no-cache-dir -i https://pypi.tuna.tsinghua.edu.cn/simple

 COPY . .

+RUN python acge_embedding.py
+
 CMD ["python", "main.py"]