本文为您介绍如何将 Stable Diffusion 使用的自定义模型上传至函数挂载的 NAS 目录进行存储，并在启动 WebUI 时实时加载该模型。

场景介绍

用户在使用 Stable Diffusion 时，可能会有使用自定义模型或 WebUI 插件的需求。由于这些模型通常较大，为了方便快速地读取，将其存储在 NAS 挂载目录中是一个常见的选择。

前提条件

• 私有网络 VPC
  • 已开通火山引擎私有网络。
  • 已在函数待部署地域，创建私有网络、子网和安全组。详细操作请参见 创建私有网络、创建子网、创建安全组。
• NAS 文件存储
  • 已开通火山引擎 NAS 文件存储。
  • 已在函数待部署地域，创建 NAS 文件系统。详细操作请参见 创建文件系统。
  • 已在目标 VPC 中创建挂载点。详细操作请参见 创建挂载点。
• 函数服务
  • 已开通火山引擎函数服务。
  • 已部署 AI 绘画 Stable Diffusion 应用。详细操作请参见 通过函数服务快速部署 AI 绘画 Stable Diffusion 应用。

操作步骤

步骤一：配置 NAS 挂载目录

1. 登录 函数服务控制台。

2. 在顶部导航栏，选择目标地域。

3. 在 函数列表 页面，选择前提条件中部署好的 Stable Diffusion 函数，单击进入当前函数的基础信息页面。

4. 在 VPC 网络区域，单击编辑图标，启用并配置 VPC 网络。
   配置函数要绑定的私有网络 VPC、子网和安全组。其中，函数配置的 VPC 必须与 NAS 挂载点的 VPC 一致。
   

5. 在存储配置区域，单击编辑图标，启用并配置 NAS 挂载目录。
   

   • NAS 文件系统：选择同地域的 NAS 文件系统。
   • 挂载点：选择与函数 VPC 匹配的 NAS 挂载点。
   • 远端目录：配置要访问的远端 NAS 的目录。本示例为：/。
   • 本地目录：配置函数运行环境的本地目录。本地目录与远端目录为映射关系。本示例为：/mnt/nas。

步骤二：配置环境变量

通过配置CLI_ARGS环境变量，将 WebUI 默认加载的模型指向 NAS 挂载的文件路径。

1. 在基础信息页面，单击配置信息区域旁的编辑图标。

2. 在编辑配置信息面板中，单击＋，新增一个环境变量。

   

   • key：固定为CLI_ARGS。
   • value：本示例使用--ckpt来指定 WebUI 启动时需要加载的 Stable Diffusion 模型文件。假设自定义模型在 NAS 中的存储路径为/models/Stable-diffusion/sd_xl_base_1.0_0.9vae.safetensors ，则对应到函数运行环境的本地目录为/mnt/nas/models/Stable-diffusion/sd_xl_base_1.0_0.9vae.safetensors。因此，value 为--ckpt /mnt/nas/models/Stable-diffusion/sd_xl_base_1.0_0.9vae.safetensors。

步骤三：发布函数

将最新的函数配置发布至线上。

1. 在函数详情页面，单击右上角的 发布。

2. 函数版本 使用 Latest，按需填写 版本描述。

   

3. 单击 确定，函数开始发布。
   函数发布成功后，便可通过 API 网关触发器提供的公网访问地址，访问 Stable Diffusion WebUI。