开发一个训练LORA的WebUI

文章目录

效果
原理
界面参数说明
安装步骤
代码

效果

在这里插入图片描述

原理

基于开源项目 kohya-ss/sd-scripts增加了Gradio UI，精简了训练参数，更适合初级“炼丹宝宝”的炼丹炉！
核心思想还是比较简单的，通过Gradio UI来收集设置的训练参数，并通过python的 subprocess 库来执行命令 accelerate launch --xxx，最终达到训练LORA的效果。

界面参数说明

基础模型：后缀为.ckpt或者.safetensors的模型，也可以指定huggingface中的模型id（需要可以顺利访问huggingface）

数据集：训练数据集

相关阅读:
Visual Studio 和 VSCode哪个更好?
Kubernetes学习笔记-一
springboot基于点餐码二维码在线点餐系统vue.js+java
学信息系统项目管理师第4版系列13_立项管理
数据库实践 Hw09
dubbo学习资料
【小米】二维定长滑动窗口
【深度学习实验】卷积神经网络（一）：卷积运算及其Pytorch实现（一维卷积：窄卷积、宽卷积、等宽卷积；二维卷积）
vue3学习（十二）--- 自定义指令
线性代数（二）| 行列式性质求值特殊行列式加边法归纳法等多种方法

原文地址：https://blog.csdn.net/weixin_40660408/article/details/133000662