File size: 598 Bytes
4161fe6
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
#!/bin/bash

# 确保所有目录存在且有正确权限
mkdir -p /app/models
chmod 777 /app/models

mkdir -p /app/.cache/huggingface/hub
chmod 777 /app/.cache/huggingface/hub

mkdir -p /tmp/iopaint/models
chmod 777 /tmp/iopaint/models

# 设置环境变量
export TRANSFORMERS_CACHE=/app/.cache/huggingface/hub
export HF_HOME=/app/.cache/huggingface
export IOPAINT_MODEL_DIR=/app/models

# 打印一些调试信息
echo "Current directory: $(pwd)"
echo "Directory permissions:"
ls -la /app/models
ls -la /app/.cache/huggingface
ls -la /tmp/iopaint

# 启动应用
exec python -m iopaint.api_only