fix: type
Some checks failed
black-action / runner / black formatter (push) Failing after 0s

This commit is contained in:
0xMRTT 2023-06-21 10:52:48 +02:00
parent 32777cca5c
commit e72ffd6d8b
Signed by: 0xMRTT
GPG Key ID: 910B287304120902

View File

@ -578,7 +578,7 @@ class AsyncImagine:
async def sdprem(self, prompt: str, negative: str = None, priority: str = None, steps: str = None,
high_res_results: str = None, style: Style = Style.IMAGINE_V1, seed: str = None,
ratio: Ratio = Ratio.RATIO_1X1, cfg: float = 9.5) -> bytes | None:
ratio: Ratio = Ratio.RATIO_1X1, cfg: float = 9.5) -> bytes:
"""Generates AI Art."""
try:
validated_cfg = validate_cfg(cfg)
@ -615,7 +615,7 @@ class AsyncImagine:
else:
return None
async def upscale(self, image: bytes) -> bytes | None:
async def upscale(self, image: bytes) -> bytes:
"""Upscales the image."""
try:
async with self.session.post(