refactor(cat-bodhi): deploy pinned upstream source

This commit is contained in:
2026-08-14 00:15:22 +08:00
parent 207fc4cc2f
commit 3a341e7a79
6 changed files with 44 additions and 23 deletions
+13 -8
View File
@@ -2,22 +2,24 @@ services:
cat-bodhi:
build:
context: .
# GPU 加速:取消下面注释(需先安装 nvidia-container-toolkit
# args:
# TORCH_INDEX: https://download.pytorch.org/whl/cu126
args:
CAT_BODHI_REF: eaf9d27b4ebe3821da983106a78aa1c745532527
CAT_BODHI_ARCHIVE_SHA256: 73257f7c9832dd06f958e43cc41d1dccff91788fe49d5384c104d24c75862d4d
# GPU 加速:取消下面注释(需先安装 nvidia-container-toolkit
# TORCH_INDEX: https://download.pytorch.org/whl/cu126
container_name: cat-bodhi
restart: unless-stopped
env_file:
- .env
environment:
- PORT=8080
- SPRITE_SEG_ROOT=/opt/sprite_alpha_seg_pytorch
- SPRITE_SEG_ROOT=/app/sprite_alpha_seg_pytorch
- SPRITE_SEG_PYTHON=/usr/local/bin/python
- SPRITE_SEG_CHECKPOINT=/opt/sprite_alpha_seg_pytorch/checkpoints/unet_sprite_ft.pt
- SPRITE_SEG_OUT_DIR=/opt/sprite_alpha_seg_pytorch/outputs/cat_match
- SPRITE_SEG_CHECKPOINT=/app/sprite_alpha_seg_pytorch/checkpoints/unet_sprite_ft.pt
- SPRITE_SEG_OUT_DIR=/app/sprite_alpha_seg_pytorch/outputs/cat_match
volumes:
# 模型文件目录(需要手动放入 infer_spritesheet_hybrid.py + checkpoints/
- ./sprite_alpha_seg_pytorch:/opt/sprite_alpha_seg_pytorch:rw
# 模型与推理代码来自固定的上游提交,只有生成结果需要持久化。
- ./data/sprite-output:/app/sprite_alpha_seg_pytorch/outputs:rw
# AI 生成图片持久化(可选,不挂载则容器重启后丢失)
- ./data/assets-ai:/app/assets/ai:rw
networks:
@@ -66,6 +68,9 @@ services:
- "xiteng.site.component.cat-bodhi.navigation=new-tab"
- "xiteng.site.component.cat-bodhi.portal-link=embedded"
- "org.opencontainers.image.source=https://github.com/ziyuezhou1/CAT-BODHI"
- "org.opencontainers.image.revision=eaf9d27b4ebe3821da983106a78aa1c745532527"
networks:
homelab_net:
external: true