fiewolf1000 commited on
Commit
d863826
·
verified ·
1 Parent(s): 849bc37

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -2
Dockerfile CHANGED
@@ -18,8 +18,9 @@ RUN pip install --no-cache-dir -r requirements.txt
18
  COPY . .
19
 
20
  # 6. 设置环境变量(可选,用于指定模型加载设备、日志级别等)
21
- ENV PYTHONUNBUFFERED=1 \
22
- TORCH_DEVICE=cuda # 若 Spaces 分配 GPU 则自动使用,无 GPU 会 fallback 到 CPU
 
23
 
24
  # 7. 暴露端口(Hugging Face Spaces 会自动映射该端口,需与 app.py 中 uvicorn 端口一致)
25
  EXPOSE 7860
 
18
  COPY . .
19
 
20
  # 6. 设置环境变量(可选,用于指定模型加载设备、日志级别等)
21
+ # 说明:PYTHONUNBUFFERED=1 确保日志实时输出;TORCH_DEVICE 仅为标识,实际以代码检测为准
22
+ ENV PYTHONUNBUFFERED=1
23
+ ENV TORCH_DEVICE=cuda
24
 
25
  # 7. 暴露端口(Hugging Face Spaces 会自动映射该端口,需与 app.py 中 uvicorn 端口一致)
26
  EXPOSE 7860