stable diffusion api 實現(xiàn)多人列隊及切換模型的方法
一、在api接口中實現(xiàn)任務(wù)列隊及獲取任務(wù)狀態(tài)的方法：? ?
1.? 在processing.py /StableDiffusionProcessingTxt2Img? ?添加id_task 參數(shù)
2. 在 processing.py /StableDiffusionProcessingImg2Img? ?添加id_task 參數(shù)
3. 修改api.py /text2imgapi：
? ? ?獲取id_task，并添加到任務(wù)列隊
4. 修改api.py /img2imgapi：
? ? ?獲取id_task，并添加到任務(wù)列隊
代碼實現(xiàn)如下圖：
以上兩個圖片，是txt2txtapi 接口文件的完整代碼，里面包含了，對任務(wù)的id記錄，和切換模型的代碼，具體實現(xiàn)思路是，在請求接口時，增加任務(wù)id,和模型參數(shù)，請求該接口后，根據(jù)傳入的模型名稱加載模型，然后把改任務(wù)加入列隊，并開始記錄任務(wù)id,繪圖結(jié)束后，結(jié)束任務(wù)ID的記錄，輸出圖像，從而實現(xiàn)了，在接口調(diào)用中加載模型和任務(wù)的id記錄。
標(biāo)簽：
