yujuanqin/test_mlx

创建环境

git clone git@hf.co:yujuanqin/test_mlx
cd test_mlx
git clone git@hf.co:mvp1024/mlx-app-pack mlx_app # 需要huggingface mvp1024 权限
conda create test_mlx python=3.10
conda activate test_mlx
pip install -r mlx_app/requirements.txt

需要修改 mlx_app 中的一些代码以避免网络请求：

• mlx_app/stable_diffusion/model_io.py: 将所有使用 hf_hub_download 的代码注释掉，改为使用 get_file_path; 修改 get_file_path 里的路径return os.path.join("../models", key, val)
• mlx_app/txt2image_lora.py: 88 行，改为unet_config = f"../models/{repo_id}/{_MODELS[repo_id]['unet_config']}"

下载模型

python download_models.py

运行测试脚本

python test_mlx.py

收集测试结果

测试结果在 result_mlx.csv,生成的图片保存在当前目录下的base_model_images文件夹。