在AutoDL云环境上训练Stable Diffusion Lora模型

培根芝士2023-11-14 14:39

点击租用新实例，并且选择你所需要的显卡。

镜像选择"社区镜像"，选择lora-scripts镜像，点击立即创建。

在"容器实例"中找到刚刚租用的实例，开机后，点击"JupyterLab"，进入目录"/lora-scripts/"，打开终端。

使用命令开启GUI

bash 复制代码

bash run_gui.sh

开启的GUI端口是28000

这里需要将实例的端口代理到本地，才能成功进入GUI界面。

在容器实例页面复制实例的SSH登录指令和密码

指令格式为：

bash 复制代码

ssh -p 35394 [email protected]

注意35394为端口号，region-1.autodl.com为远程地址，请更换为你的实例端口和地址。

在本地终端输入命令：

bash 复制代码

ssh -CNg -L 28000:127.0.0.1:28000 [email protected] -p 35394

输入密码，就可以在本地访问端口了，成功进入GUI

bash 复制代码

http://127.0.0.1:28000

修改数据集设置和保存设置，配置训练数据集路径、模型保存名称、模型保存文件夹。

在GUI中配置好参数点击"开始训练"即可开始训练。

数据盘对应的实例目录是/root/autodl-fs

直接使用阿里云盘、百度网盘上传下载数据，强烈推荐

和网盘类似，优点简单，但是只支持文件不支持文件夹

支持文件和文件夹上传下载

上传指令为：

bash 复制代码

scp -rP 35394 <本地文件/文件夹> [email protected]:/root/lora-scripts/

下载指令为：

bash 复制代码

scp -rP 35394 [email protected]:/root/lora-scripts/output <本地文件/文件夹>

支持文件和文件夹，拖拉拽即可上传下载，推荐使用