初次使用 minimind

前言

一直在探索市面上轻量化的个人 AI 模型，但是目前像 ds、chatgpt、kimi…都太过于巨大，训练要求也太过于高，所以一直在找是不是有一款阉割版的模型可以提供给个人普通电脑来使用，并且训练。

有的，兄弟有的。

这就是本文需要介绍的 minimind 模型。

安装

minimind 模型的安装非常简单，只需要安装好 python 环境和相关的依赖即可。

由于 minimind 模型的依赖比较多，所以建议使用 conda 环境来安装。严格按照官方推荐的 py 版本会减少很多不必要的麻烦。

依次执行以下命令：

0. 创建 conda 环境（可选）

如果没有安装 conda 环境，建议先安装好 conda 环境。conda官方网站, 安装结束后运行以下命令：

conda create -n minimind python=3.10.16
conda activate minimind

python --version # 查看 py 版本是否为 3.10.16

当然如果你有其它更好的方案，或者想体验这个解决问题的过程，当然也可以跳过这一步。

1. 克隆代码

git clone https://github.com/jingyaogong/minimind.git

2. 安装依赖

记住这里需要进入项目根目录下安装

cd minimind
pip install -r requirements.txt -i https://mirrors.aliyun.com/pypi/simple

3. 下载模型

git clone https://huggingface.co/jingyaogong/MiniMind2 # or https://www.modelscope.cn/models/gongjy/MiniMind2

注意：这里有可能 model.safetensors 文件下载失败，这是因为 huggingface.co 上的文件比较大，下载速度比较慢。如果下载失败，建议使用 modelscope.cn 上的模型。或者直接下载模型文件手动添加到当前目录。

4. 运行 web 界面

cd scripts
streamlit run web_demo.py