44 lines
1.6 KiB
Python
44 lines
1.6 KiB
Python
|
from .base import AIDRAW_BASE
|
|||
|
|
from ..config import config
|
|||
|
|
|
|||
|
|
|
|||
|
|
class AIDRAW(AIDRAW_BASE):
|
|||
|
|
"""队列中的单个请求"""
|
|||
|
|
max_resolution: int = 32
|
|||
|
|
|
|||
|
|
async def fromresp(self, resp):
|
|||
|
|
img: dict = await resp.json()
|
|||
|
|
return img["images"][0]
|
|||
|
|
|
|||
|
|
async def post(self):
|
|||
|
|
site=config.novelai_site or "127.0.0.1:7860"
|
|||
|
|
header = {
|
|||
|
|
"content-type": "application/json",
|
|||
|
|
"user-agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/106.0.0.0 Safari/537.36",
|
|||
|
|
}
|
|||
|
|
post_api = f"http://{site}/sdapi/v1/img2img" if self.img2img else f"http://{site}/sdapi/v1/txt2img"
|
|||
|
|
for i in range(self.batch):
|
|||
|
|
parameters = {
|
|||
|
|
"prompt": self.tags,
|
|||
|
|
"seed": self.seed[i],
|
|||
|
|
"steps": self.steps,
|
|||
|
|
"cfg_scale": self.scale,
|
|||
|
|
"width": self.width,
|
|||
|
|
"height": self.height,
|
|||
|
|
"sampler_name": self.sampler,
|
|||
|
|
"negative_prompt": self.ntags,
|
|||
|
|
"enable_hr": self.hires,
|
|||
|
|
"denoising_strength": 0.7,
|
|||
|
|
"hr_scale": 2,
|
|||
|
|
"hr_upscaler": "Latent"
|
|||
|
|
}
|
|||
|
|
print("向API发送以下参数：")
|
|||
|
|
print(parameters)
|
|||
|
|
if self.img2img:
|
|||
|
|
parameters.update({
|
|||
|
|
"init_images": ["data:image/jpeg;base64,"+self.image],
|
|||
|
|
"denoising_strength": self.strength,
|
|||
|
|
})
|
|||
|
|
await self.post_(header, post_api, parameters)
|
|||
|
|
return self.result
|