Update Dockerfile

Dockerfile

@@ -9,18 +9,18 @@ RUN apt-get update && apt-get install -y \
     curl \
     && rm -rf /var/lib/apt/lists/*
 
-# 设置缓存目录并创建目录，使用 HF_HOME
-ENV HF_HOME=/root/.cache/huggingface
+# 设置缓存目录并使用 HF_HOME 环境变量（推荐使用 HF_HOME）
+ENV HF_HOME=/tmp/huggingface_cache
 
 # 创建缓存目录并设置权限
-RUN mkdir -p /root/.cache/huggingface && chmod -R 777 /root/.cache/huggingface
+RUN mkdir -p /tmp/huggingface_cache && chmod -R 777 /tmp/huggingface_cache
 
 # 创建并进入工作目录
 WORKDIR /app
 
 # 安装依赖
 RUN pip install --upgrade pip
-RUN pip install torch transformers peft datasets accelerate bitsandbytes
+RUN pip install torch transformers peft datasets accelerate bitsandbytes huggingface_hub
 
 # 将代码和数据复制到容器中
 COPY . /app