blob: c1ddad93f25620b8e77f238f2d46eb69f125ca3c (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
|
from __future__ import annotations
from ..schema import GenerationResult, TextToImageRequest
from ..service import ImageGenerationService
async def generate_image(
service: ImageGenerationService,
positive_prompt: str,
negative_prompt: str | None = None,
size: str | None = None,
) -> GenerationResult:
request = TextToImageRequest.create(
positive_prompt,
negative_prompt,
size,
default_size=service.config.generation.default_size,
base_positive=service.config.prompt.positive,
base_negative=service.config.prompt.negative,
)
return await service.generate_image(request)
|
