Immich 机器学习模型手动下载与配置
目录
immich 的机器学习任务执行前，需要从 Huggingface 下载需要使用的大模型，但是因为国内网络问题，无法访问 Huggingface，导致机器学习相关的任务执行失败。接下来我们来解决这个问题。
步骤一：下载大模型 #
- 安装 git-lfs
# macOS
brew install git-lfs
# Debian
sudo apt install git-lfs
- 如果是在国内的话，需要开启科学上网，然后下载需要使用的两个模型文件，时间较久，耐心等待下载完成
# clone buffalo_l
git clone https://huggingface.co/immich-app/buffalo_l
# clone ViT-B-32__openai
git clone https://huggingface.co/immich-app/ViT-B-32__openai
# 进入 buffalo_l目录
cd buffalo_l
# 拉取buffalo_l大模型文件
git lfs install
git lfs pull
# 返回上层并进入ViT-B-32__openai
cd ../ViT-B-32__openai
# 拉取ViT-B-32__openai大模型文件
git lfs install
git lfs pull
# 返回上级目录
cd ..
步骤二：上传到 immich 的 Docker 容器内 #
- 进入容器，并创建文件夹，然后退出容器
# 进入immich_machine_learning容器
sudo docker exec -it immich_machine_learning /bin/bash
# 创建大模型存放文件夹
mkdir -p /cache/clip
mkdir -p /cache/facial-recognition
# 退出容器
exit
- 拷贝大模型到容器内
# 拷贝大模型到容器内
docker cp ViT-B-32__openai immich_machine_learning:/cache/clip/
docker cp buffalo_l immich_machine_learning:/cache/facial-recognition/
# 授权
docker exec immich_machine_learning chmod -R 777 /cache
# 重启容器
sudo docker restart immich_machine_learning
步骤三：验证 #
打开 immich，运行机器学习相关的任务，查看日志或者查看 docker 状态，可以看到 immich_machine_learning 容器的 CPU 使用率已经开始升高，说明任务开始运行。
docker stats