24 lines
1.2 KiB
Plaintext
24 lines
1.2 KiB
Plaintext
# ============================================================================
|
||
# BiRefNet 抠图服务依赖清单
|
||
# 安装：pip install -r requirements.txt
|
||
#
|
||
# 注意 1：PyPI 默认安装的是 CPU 版 torch。
|
||
# 如果你的机器有 NVIDIA GPU，请先按 PyTorch 官网命令单独安装对应 CUDA 版 torch，
|
||
# 再执行 pip install -r requirements.txt（已安装的 torch/torchvision 会被跳过）：
|
||
# pip install torch torchvision --index-url https://download.pytorch.org/whl/cu121
|
||
#
|
||
# 注意 2：模型会自动下载到 HuggingFace 默认缓存目录 ~/.cache/huggingface（走 HF_ENDPOINT 镜像）。
|
||
# 首次启动若本地无缓存会联网下载 BiRefNet_lite；之后启动命中本地缓存则完全离线。
|
||
# ============================================================================
|
||
|
||
torch>=2.0.0
|
||
torchvision>=0.15.0
|
||
transformers>=4.44.0
|
||
fastapi>=0.100.0
|
||
uvicorn>=0.23.0
|
||
pillow>=9.5.0
|
||
python-multipart>=0.0.9 # FastAPI 处理 multipart 表单（UploadFile）必需
|
||
einops>=0.7.0 # BiRefNet 自定义建模代码（birefnet.py）依赖
|
||
kornia>=0.7.0 # BiRefNet 自定义建模代码（birefnet.py）依赖
|
||
timm>=0.9.0 # BiRefNet 自定义建模代码（birefnet.py）依赖
|