immich机器学习AI大模型配置

immich的机器学习任务执行前，需要从Huggingface下载需要使用的大模型，但是因为国内网络问题，无法访问Huggingface，导致机器学习相关的任务执行失败。接下来我们来解决这个问题。

步骤一：下载大模型

1. 安装git-lfs​

   # macOS
   brew install git-lfs
   
   # Debian
   sudo apt install git-lfs

2. 如果是在国内的话，需要开启科学上网，然后下载需要使用的两个模型文件，时间较久，耐心等待下载完成

   # 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容器内

1. 进入容器，并创建文件夹，然后退出容器

   # 进入immich_machine_learning容器
   sudo docker exec -it immich_machine_learning /bin/bash
   # 创建大模型存放文件夹
   mkdir -p /cache/clip
   mkdir -p /cache/facial-recognition
   # 退出容器
   exit

2. 拷贝大模型到容器内

   # 拷贝大模型到容器内
   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

参考文档：IMMICH: 手动下载AI大模型并放置在正确位置