stable diffusion api 实现多人列队及切换模型的方法
一、在api接口中实现任务列队及获取任务状态的方法:
1. 在processing.py /StableDiffusionProcessingTxt2Img 添加id_task 参数
2. 在 processing.py /StableDiffusionProcessingImg2Img 添加id_task 参数
3. 修改api.py /text2imgapi:
获取id_task,并添加到任务列队
4. 修改api.py /img2imgapi:
获取id_task,并添加到任务列队
代码实现如下图:


以上两个图片,是txt2txtapi 接口文件的完整代码,里面包含了,对任务的id记录,和切换模型的代码,具体实现思路是,在请求接口时,增加任务id,和模型参数,请求该接口后,根据传入的模型名称加载模型,然后把改任务加入列队,并开始记录任务id,绘图结束后,结束任务ID的记录,输出图像,从而实现了,在接口调用中加载模型和任务的id记录。