Akatuki25 commited on
Commit
6a0b6dd
·
1 Parent(s): 38fbde8

Fix checkpoint directory permissions and unify env vars

Browse files
Files changed (1) hide show
  1. Dockerfile +5 -3
Dockerfile CHANGED
@@ -3,7 +3,9 @@ FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04
3
  # 環境変数
4
  ENV DEBIAN_FRONTEND=noninteractive
5
  ENV PYTHONUNBUFFERED=1
6
- ENV HF_HUB_CACHE=/app/checkpoints/hf_cache
 
 
7
 
8
  # 作業ディレクトリ
9
  WORKDIR /app
@@ -25,8 +27,8 @@ RUN pip3 install --no-cache-dir -r requirements.txt
25
  COPY app.py .
26
  COPY seed-vc/ ./seed-vc/
27
 
28
- # チェックポイント用ディレクトリ作成
29
- RUN mkdir -p /app/checkpoints/hf_cache
30
 
31
  # ポート公開(HF Spacesは7860を使用)
32
  EXPOSE 7860
 
3
  # 環境変数
4
  ENV DEBIAN_FRONTEND=noninteractive
5
  ENV PYTHONUNBUFFERED=1
6
+ ENV HF_HOME=/app/checkpoints
7
+ ENV HF_HUB_CACHE=/app/checkpoints
8
+ ENV TRANSFORMERS_CACHE=/app/checkpoints
9
 
10
  # 作業ディレクトリ
11
  WORKDIR /app
 
27
  COPY app.py .
28
  COPY seed-vc/ ./seed-vc/
29
 
30
+ # チェックポイント用ディレクトリ作成(書き込み権限を確保)
31
+ RUN mkdir -p /app/checkpoints && chmod -R 777 /app/checkpoints
32
 
33
  # ポート公開(HF Spacesは7860を使用)
34
  EXPOSE 7860