欢迎光临散文网 会员登陆 & 注册

stable diffusion api 实现多人列队及切换模型的方法

2023-07-19 20:06 作者:不是菜鳥  | 我要投稿

一、在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记录。

stable diffusion api 实现多人列队及切换模型的方法的评论 (共 条)

分享到微博请遵守国家法律