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

2024-01-26 11:37:13 +08:00
parent 766ac8704a
commit 773eb8a1bf
3 changed files with 11 additions and 2 deletions
@@ -1,2 +1,3 @@
 __pycache__
-/venv
+/venv
+/huggingface
@@ -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"]
@@ -0,0 +1,6 @@
+#!/bin/bash
+
+export HF_HOME="$(pwd)/huggingface"
+echo HF_HOME is $HF_HOME
+source venv/bin/activate
+python3 main.py